use importlib_metadata backport on PyPI

scikit-hep · Aug 16, 2023 · 9ada877 · 9ada877
1 parent e37fc3f
commit 9ada877
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -52,6 +52,7 @@ dependencies = [
     "scipy>=1.2.0",  # requires numpy, which is required by pyhf and tensorflow
     "tqdm>=4.56.0",  # for readxml
     "typing_extensions>=3.7.4.3; python_version == '3.7'",  # for SupportsIndex
+    "importlib-metadata>=0.1; python_version == '3.7'",  # for importlib.metadata.version
     "numpy",  # compatible versions controlled through scipy
 ]
 

diff --git a/src/pyhf/tensor/jax_backend.py b/src/pyhf/tensor/jax_backend.py
@@ -15,15 +15,11 @@
 # v0.7.x backport hack
 try:
     from importlib.metadata import version
-
-    _jax__version__ = version("jax")
 except ModuleNotFoundError:
     # importlib.metadata added in Python 3.8
-    import jax
+    from importlib_metadata import version
 
-    _jax__version__ = jax.__version__
-_jax__version__ = tuple(map(int, (_jax__version__.split("."))))
-_old_jax_version = _jax__version__ < (0, 4, 1)
+_old_jax_version = tuple(map(int, (version("jax").split(".")))) < (0, 4, 1)
 
 if not _old_jax_version:
     from jax import Array

diff --git a/tests/constraints.txt b/tests/constraints.txt
@@ -7,6 +7,7 @@ jsonpatch==1.15
 pyyaml==5.1
 importlib_resources==1.4.0  # c.f. PR #1979
 typing-extensions==3.7.4.3  # c.f. PR #1961, #1940
+importlib-metadata==0.1  # c.f. PR #2280
 # xmlio
 uproot==4.1.1
 # minuit