MNT: Use importlib.metadata instead of pkg_resources

Use PEP-440 compatible fallback

Author: effigies

templateflow/__init__.py

@@ -6,14 +6,13 @@
 try:
     from ._version import __version__
 except ModuleNotFoundError:
-    from pkg_resources import get_distribution, DistributionNotFound
-
+    from importlib.metadata import version, PackageNotFoundError
     try:
-        __version__ = get_distribution(__packagename__).version
-    except DistributionNotFound:
-        __version__ = "unknown"
-    del get_distribution
-    del DistributionNotFound
+        __version__ = version(__packagename__)
+    except PackageNotFoundError:
+        __version__ = "0+unknown"
+    del version
+    del PackageNotFoundError
 
 import os
 from . import api

templateflow/tests/test_version.py

@@ -1,7 +1,7 @@
 """Test _version.py."""
 import sys
 from collections import namedtuple
-from pkg_resources import DistributionNotFound
+from importlib.metadata import PackageNotFoundError
 from importlib import reload
 import templateflow
 
@@ -18,14 +18,13 @@ class _version:
 
 
 def test_version_scm1(monkeypatch):
-    """Retrieve the version via pkg_resources."""
+    """Retrieve the version via importlib.metadata."""
     monkeypatch.setitem(sys.modules, "templateflow._version", None)
 
-    def _dist(name):
-        Distribution = namedtuple("Distribution", ["name", "version"])
-        return Distribution(name, "success")
+    def _ver(name):
+        return "success"
 
-    monkeypatch.setattr("pkg_resources.get_distribution", _dist)
+    monkeypatch.setattr("importlib.metadata.version", _ver)
     reload(templateflow)
     assert templateflow.__version__ == "success"
 
@@ -35,8 +34,8 @@ def test_version_scm2(monkeypatch):
     monkeypatch.setitem(sys.modules, "templateflow._version", None)
 
     def _raise(name):
-        raise DistributionNotFound("No get_distribution mock")
+        raise PackageNotFoundError("No get_distribution mock")
 
-    monkeypatch.setattr("pkg_resources.get_distribution", _raise)
+    monkeypatch.setattr("importlib.metadata.version", _raise)
     reload(templateflow)
-    assert templateflow.__version__ == "unknown"
+    assert templateflow.__version__ == "0+unknown"