From 6f39c79a2756a9779e7e6f83488da0bfa2610868 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?R=C3=A9mi=20Flamary?= Date: Thu, 29 Feb 2024 16:07:04 +0100 Subject: [PATCH 1/2] Update requirements.txt Limit the max version of jax to allow for old config because of pymanopt using deprecated version --- requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index 6af50f127..3301c94e3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,11 +6,11 @@ pymanopt cvxopt scikit-learn torch -jax +jax<=0.4.24 jaxlib tensorflow pytest torch_geometric cvxpy geomloss -pykeops \ No newline at end of file +pykeops From e7688a010d504baf5aac8f0ca8dadaa1504b7aaf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?R=C3=A9mi=20Flamary?= Date: Thu, 29 Feb 2024 16:14:14 +0100 Subject: [PATCH 2/2] Update requirements.txt --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 3301c94e3..bffaf892f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,7 +7,7 @@ cvxopt scikit-learn torch jax<=0.4.24 -jaxlib +jaxlib<=0.4.24 tensorflow pytest torch_geometric