From ebcdedf109b4ab1c287ac527b59493e08f097b4f Mon Sep 17 00:00:00 2001
From: Colin Carroll <colcarroll@google.com>
Date: Tue, 19 Dec 2023 11:12:53 -0800
Subject: [PATCH] Update requirements to allow TFP >= 0.20. Also removes
 unnecessary typing_extensions.

PiperOrigin-RevId: 592290386
---
 pyproject.toml | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index ebc1183..5d8b432 100644
--- 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]