From ba0909aa243bacd5ecf6a32e790a4713bfaa2fa0 Mon Sep 17 00:00:00 2001 From: Krzysztof Rusek Date: Wed, 29 Oct 2025 10:46:12 +0100 Subject: [PATCH 1/3] Update dependency version constraints in pyproject.toml --- pyproject.toml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 48e9414..71b2f54 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -17,9 +17,9 @@ authors = [ requires-python = ">=3.12" dependencies = [ - "chex~=0.1.88", - "jax~=0.4.38", - "tensorflow-probability[jax]~=0.25.0" + "chex>=0.1.88", + "jax>=0.4.38", + "tensorflow-probability[jax]>=0.25.0" ] [build-system] @@ -28,5 +28,5 @@ build-backend = "setuptools.build_meta" [project.optional-dependencies] graph = [ - "jraph~=0.0.6.dev0" + "jraph>=0.0.6.dev0" ] From 02b396e7b1d24f77ca63432797b91df02ae2a0d8 Mon Sep 17 00:00:00 2001 From: Krzysztof Rusek Date: Wed, 29 Oct 2025 11:04:09 +0100 Subject: [PATCH 2/3] Try to fix ci --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 71b2f54..38d1b0d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -19,7 +19,7 @@ requires-python = ">=3.12" dependencies = [ "chex>=0.1.88", "jax>=0.4.38", - "tensorflow-probability[jax]>=0.25.0" + "tensorflow-probability[jax]" ] [build-system] From e1b34067e984d982e156c19057f63de961614270 Mon Sep 17 00:00:00 2001 From: Krzysztof Rusek Date: Wed, 29 Oct 2025 11:14:07 +0100 Subject: [PATCH 3/3] Try to fix ci v2 --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 38d1b0d..b0e3aa7 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -19,7 +19,7 @@ requires-python = ">=3.12" dependencies = [ "chex>=0.1.88", "jax>=0.4.38", - "tensorflow-probability[jax]" + "tfp-nightly[jax]" ] [build-system]