diff --git a/python/requirements/ml/requirements_dl.txt b/python/requirements/ml/requirements_dl.txt index 3970fa10a975..e399ae78d02a 100644 --- a/python/requirements/ml/requirements_dl.txt +++ b/python/requirements/ml/requirements_dl.txt @@ -8,7 +8,7 @@ # TODO(amogkam): Remove after https://github.com/tensorflow/tensorflow/issues/52922 is fixed. keras==2.6.0 tensorflow==2.6.0 -tensorflow-probability==0.14.0 +tensorflow-probability==0.15.0 torch==1.9.0;sys_platform=="darwin" torchvision==0.10.0;sys_platform=="darwin"