use tensorflow_probability[tf] extra to match versions

Fixes #826. Related to #434 and possibly #727.
dianna-ai · Sep 19, 2024 · 8af438c · 8af438c
1 parent f742047
commit 8af438c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -51,8 +51,7 @@ install_requires =
     xarray
     torch <=2.3.1  # this is temporary, see #829
     # 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'