From feddd28a63d647ad10c9403eb91a23720320e455 Mon Sep 17 00:00:00 2001
From: Michael Reneer <michaelreneer@google.com>
Date: Fri, 23 Sep 2022 13:04:14 -0700
Subject: [PATCH] Pin `tensorflow-probability` to a specific version.

This matches the documentation for specifying version for this project. Additionally, recent version of `tensorflow-probability` are releasing breaking changes and the existing version specifier allows these to breaking changes to be unintentionally pulled in.

PiperOrigin-RevId: 476443999
---
 requirements.txt | 2 +-
 setup.py         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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())