use tensorflow_probability[tf] extra to match versions

Fixes #826. Related to #434 and possibly #727.
dianna-ai · Jul 29, 2024 · 24f3d05 · 24f3d05
1 parent b9faca7
commit 24f3d05
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -50,8 +50,7 @@ install_requires =
     tqdm
     xarray
     # tf and tfprob are required but not declared by onnx_tf
-    tensorflow >= 2.12,<2.16
-    tensorflow_probability <0.24
+    tensorflow_probability[tf]
     # https://setuptools.pypa.io/en/latest/userguide/datafiles.html#accessing-data-files-at-runtime
     importlib_resources;python_version<'3.10'