Update requirements to allow TFP >= 0.20. Also removes unnecessary ty…

…ping_extensions. PiperOrigin-RevId: 592290386
jax-ml · Dec 19, 2023 · ebcdedf · ebcdedf
1 parent 6562883
commit ebcdedf
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,8 +8,7 @@ license = "Apache 2.0"
 [tool.poetry.dependencies]
 python = "^3.9"
 jax = "^0.4.6"
-tensorflow-probability = "^0.19.0"
-typing-extensions = "^4.4.0"
+tensorflow-probability = ">=0.19.0"
 
 
 [tool.poetry.dev-dependencies]