diff --git a/setup.py b/setup.py index 1e8151b7..7afded0b 100644 --- a/setup.py +++ b/setup.py @@ -41,7 +41,7 @@ "rich>=9.2.0", "numpy>=1.19.5", "scipy>=1.5.2", - "tensorflow>=2.4.1", + "tensorflow==2.14.1", "tensorflow-estimator>=2.4.0", "tensorflow-probability>=0.12.1", ],