pin tensorflow-probability version until an updated AMI is released

neuronets · Oct 3, 2023 · 1574b4f · 1574b4f
1 parent 475f5df
commit 1574b4f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.cfg b/setup.cfg
@@ -37,7 +37,7 @@ install_requires =
     nibabel
     numpy
     scikit-image
-    tensorflow-probability >= 0.11.0
+    tensorflow-probability == 0.21.0
     tensorflow == 2.13
     tensorflow-addons >= 0.12.0
     psutil