diff --git a/requirements.txt b/requirements.txt index 333fa2d..3bcbd66 100644 --- a/requirements.txt +++ b/requirements.txt @@ -32,5 +32,5 @@ scikit-learn>=1.0,==1.* scipy~=1.7 tensorflow-datasets~=4.5 tensorflow-estimator~=2.4 -tensorflow-probability~=0.15 +tensorflow-probability==0.15.0 tensorflow~=2.4 diff --git a/setup.py b/setup.py index 7bf038f..fbaccc9 100644 --- a/setup.py +++ b/setup.py @@ -38,7 +38,7 @@ setup( 'scipy~=1.7', 'tensorflow-datasets~=4.5', 'tensorflow-estimator~=2.4', - 'tensorflow-probability~=0.15', + 'tensorflow-probability==0.15.0', 'tensorflow~=2.4', ], packages=find_packages())