From a397cb6e9786e7eb63e0c301642658b79436be3b Mon Sep 17 00:00:00 2001
From: Grzegorz Latosinski <glatosinski@antmicro.com>
Date: Wed, 3 Apr 2024 00:06:18 +0200
Subject: [PATCH] requirements.txt: Added TensorFlow Probability module

Signed-off-by: Grzegorz Latosinski <glatosinski@antmicro.com>
---
 requirements.txt | 1 +
 1 file changed, 1 insertion(+)

diff --git a/requirements.txt b/requirements.txt
index 58005a6..a70c4d8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,6 +8,7 @@ Pillow
 scikit_learn
 tensorflow==2.13.0
 tensorflow_model_optimization
+tensorflow_probability==0.21.0
 tflite
 torch
 torchvision