From 49ead13a9e97e25388bc0c1ae631431187e9956d Mon Sep 17 00:00:00 2001 From: Clement Date: Wed, 11 Feb 2026 09:23:16 +0100 Subject: [PATCH 1/2] Fix test jax_version in backend --- ot/backend.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/ot/backend.py b/ot/backend.py index 6b03f5cd1..3f158d166 100644 --- a/ot/backend.py +++ b/ot/backend.py @@ -122,7 +122,12 @@ from jax.extend.backend import get_backend as _jax_get_backend jax_type = jax.numpy.ndarray - jax_new_version = float(".".join(jax.__version__.split(".")[1:])) > 4.24 + jax_new_version = tuple([float(s) for s in jax.__version__.split(".")]) > ( + 0, + 4, + 24, + 0, + ) except ImportError: jax = False jax_type = float From a603a0d7757acee433fd19e29f2f303badc0e275 Mon Sep 17 00:00:00 2001 From: Clement Date: Wed, 11 Feb 2026 09:35:17 +0100 Subject: [PATCH 2/2] Fix test jax_version in backend --- RELEASES.md | 1 + 1 file changed, 1 insertion(+) diff --git a/RELEASES.md b/RELEASES.md index 1660913cd..76514ee36 100644 --- a/RELEASES.md +++ b/RELEASES.md @@ -21,6 +21,7 @@ This new release adds support for sparse cost matrices and a new lazy EMD solver - Fix openmp flags on macOS (PR #789) - Clean documentation (PR #787) - Fix code coverage (PR #791) +- Fix test of the version of jax in `ot.backend` (PR #794) ## 0.9.6.post1