Replace deprecated pkg_resources

pybamm/util.py

@@ -18,7 +18,7 @@
 from warnings import warn
 
 import numpy as np
-import pkg_resources
+import importlib.metadata as il
 
 import pybamm
 
@@ -271,9 +271,10 @@ def have_jax():
 
 def is_jax_compatible():
     """Check if the available version of jax and jaxlib are compatible with PyBaMM"""
-    return pkg_resources.get_distribution("jax").version.startswith(
-        JAX_VERSION
-    ) and pkg_resources.get_distribution("jaxlib").version.startswith(JAXLIB_VERSION)
+    return (
+        il.distribution("jax").version.startswith(JAX_VERSION)
+        and il.distribution("jaxlib").version.startswith(JAXLIB_VERSION)
+    )
 
 
 def is_constant_and_can_evaluate(symbol):

tests/unit/test_parameters/test_parameter_sets_class.py

@@ -4,7 +4,7 @@
 from tests import TestCase
 
 import pybamm
-import pkg_resources
+import importlib.metadata as il
 import unittest
 
 
@@ -25,7 +25,7 @@ def test_all_registered(self):
         """Check that all parameter sets have been registered with the
         ``pybamm_parameter_sets`` entry point"""
         known_entry_points = set(
-            ep.name for ep in pkg_resources.iter_entry_points("pybamm_parameter_sets")
+            ep.name for ep in il.entry_points()["pybamm_parameter_sets"]
         )
         self.assertEqual(set(pybamm.parameter_sets.keys()), known_entry_points)
         self.assertEqual(len(known_entry_points), len(pybamm.parameter_sets))