diff --git a/requirements.txt b/requirements.txt index 1694e6f..d6cfb1f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,5 +9,5 @@ scipy~=1.2.1 sklearn~=1.0.2 tensorflow-datasets~=4.5.2 tensorflow-estimator>=2.3.0 -tensorflow-probability>=0.13.0 +tensorflow-probability~=0.15.0 tensorflow>=1.14 diff --git a/setup.py b/setup.py index 0f1880b..98a6f97 100644 --- a/setup.py +++ b/setup.py @@ -37,7 +37,7 @@ setup( 'sklearn~=1.0.2', 'tensorflow-datasets~=4.5.2', 'tensorflow-estimator>=2.3.0', - 'tensorflow-probability>=0.13.0', + 'tensorflow-probability~=0.15.0', ], # Explicit dependence on TensorFlow is not supported. # See https://github.com/tensorflow/tensorflow/issues/7166