From 8af438cc04e002889b1d6b60b10a2a05b56a959f Mon Sep 17 00:00:00 2001
From: "E. G. Patrick Bos" <egpbos@gmail.com>
Date: Mon, 29 Jul 2024 15:55:09 +0200
Subject: [PATCH] use tensorflow_probability[tf] extra to match versions

Fixes #826.

Related to #434 and possibly #727.
---
 setup.cfg | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/setup.cfg b/setup.cfg
index b363bd44..e1b8038c 100644
--- 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'