streamlined setup.py with new requirements

AaltoML · Dec 22, 2023 · 57b3d27 · 57b3d27
1 parent 6cd6fa1
commit 57b3d27
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -11,7 +11,7 @@
         "jax==0.4.14",
         "jaxlib==0.4.14",
         "objax==1.7.0",
-        "tensorflow_probability==0.20.1",
+        "tensorflow_probability==0.21",
         "numpy>=1.22"
     ],
     url='https://github.com/AaltoML/BayesNewton',