From 57b3d2758cef34a055cb1b54e1b51ff97c1ebf19 Mon Sep 17 00:00:00 2001
From: Thore Wietzke <thore.wietzke@fau.de>
Date: Fri, 22 Dec 2023 12:38:02 +0100
Subject: [PATCH] streamlined setup.py with new requirements

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 20b8e10..f61c6bd 100644
--- 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',