fix test environment

Onno Kampman · Onno Kampman · commit 5c722b2bbbf5 · 2024-04-07T18:28:54.000+08:00
diff --git a/setup.py b/setup.py
@@ -30,7 +30,9 @@ def read_file(filename: str) -> str:
         'rpy2==3.4.5',
         'scipy',
         'statsmodels',
-        'tensorflow>=2.10',
+        'tensorflow>=2.10,<=2.15',
+        'tensorflow-probability<=0.23',
+        'tf-keras',
     ],
     python_requires='>=3.10',
     zip_safe=False
diff --git a/tests_requirements.txt b/tests_requirements.txt
@@ -12,4 +12,5 @@ scikit-learn
 scipy
 statsmodels
 tensorflow>=2.10,<=2.15
+tensorflow-probability<=0.23
 tf-keras
