diff --git a/setup.py b/setup.py
index fd98fbd..6fb8dc3 100644
--- a/setup.py
+++ b/setup.py
@@ -51,7 +51,7 @@
         "pygame",
         "swig",
         "dm_control",
-        "tensorflow_probability==0.22.1",
+        "tensorflow_probability==0.25.0",
         "wandb",
         "pyyaml",
         "lxml",