From f9cc746e760c7336d39686ac9dba97fa75157514 Mon Sep 17 00:00:00 2001 From: Kevin Rue-Albrecht Date: Mon, 19 Feb 2024 15:00:31 +0000 Subject: [PATCH] remove version pin on jax --- pyproject.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 47bfcb60..afa75efa 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -36,8 +36,8 @@ dependencies = [ "drmaa", "gevent", "harmonypy", - "jax<=0.4.23", - "jaxlib<=0.4.23", + "jax", + "jaxlib", "leidenalg", "louvain", "matplotlib<=3.7.3",