switch from pkg_resources to importlib for version provision (#414)

open-atmos · Dec 4, 2023 · 7eec9a7 · 7eec9a7
1 parent e75dad4
commit 7eec9a7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/PyMPDATA/__init__.py b/PyMPDATA/__init__.py
@@ -8,7 +8,7 @@
 ![](https://github.com/atmos-cloud-sim-uj/PyMPDATA/releases/download/tip/readme_grid.png)
 """
 # pylint: disable=invalid-name
-from pkg_resources import DistributionNotFound, VersionConflict, get_distribution
+from importlib.metadata import PackageNotFoundError, version
 
 from .options import Options
 from .scalar_field import ScalarField
@@ -17,7 +17,7 @@
 from .vector_field import VectorField
 
 try:
-    __version__ = get_distribution(__name__).version
-except (DistributionNotFound, VersionConflict):
+    __version__ = version(__name__)
+except PackageNotFoundError:
     # package is not installed
     pass
diff --git a/examples/PyMPDATA_examples/__init__.py b/examples/PyMPDATA_examples/__init__.py
@@ -2,10 +2,10 @@
 PyMPDATA_examples package includes common Python modules used in PyMPDATA smoke tests
 and in example notebooks (but the package wheels do not include the notebooks)
 """
-from pkg_resources import DistributionNotFound, VersionConflict, get_distribution
+from importlib.metadata import PackageNotFoundError, version
 
 try:
-    __version__ = get_distribution(__name__).version
-except (DistributionNotFound, VersionConflict):
+    __version__ = version(__name__)
+except PackageNotFoundError:
     # package is not installed
     pass