RF: Remove pkg_resources dependency

effigies · effigies · commit 21ed4720f870 · 2024-03-22T21:42:32.000-04:00
diff --git a/pydra/tasks/nipype1/tests/__init__.py b/pydra/tasks/nipype1/tests/__init__.py
@@ -0,0 +1,18 @@
+import sys
+from atexit import register
+from contextlib import ExitStack
+from functools import lru_cache
+
+
+if sys.version_info < (3, 9):
+    from importlib_resources import as_file, files
+else:
+    from importlib.resources import as_file, files
+
+_stack = ExitStack()
+register(_stack.close)
+
+
+@lru_cache
+def load_resource(anchor, *parts) -> str:
+    return str(_stack.enter_context(as_file(files(anchor).joinpath(*parts))))
diff --git a/pydra/tasks/nipype1/tests/test_nipype1task.py b/pydra/tasks/nipype1/tests/test_nipype1task.py
@@ -1,6 +1,6 @@
 import pytest
 import shutil
-from pkg_resources import resource_filename
+from . import load_resource
 
 from nipype.interfaces import fsl
 import nipype.interfaces.utility as nutil
@@ -12,7 +12,7 @@
 def test_isolation(tmp_path):
     in_file = tmp_path / "orig/tpms_msk.nii.gz"
     in_file.parent.mkdir()
-    shutil.copyfile(resource_filename("nipype", "testing/data/tpms_msk.nii.gz"), in_file)
+    shutil.copyfile(load_resource("nipype", "testing/data/tpms_msk.nii.gz"), in_file)
 
     out_dir = tmp_path / "output"
     out_dir.mkdir()
diff --git a/pydra/tasks/nipype1/utils.py b/pydra/tasks/nipype1/utils.py
@@ -27,11 +27,11 @@ class Nipype1Task(pydra.engine.task.TaskBase):
     in Pydra Task outputs.
 
     >>> import pytest
-    >>> from pkg_resources import resource_filename
+    >>> from pydra.tasks.nipype1.tests import load_resource
     >>> from nipype.interfaces import fsl
     >>> if fsl.Info.version() is None:
     ...     pytest.skip()
-    >>> img = resource_filename('nipype', 'testing/data/tpms_msk.nii.gz')
+    >>> img = load_resource('nipype', 'testing/data/tpms_msk.nii.gz')
 
     >>> from pydra.tasks.nipype1.utils import Nipype1Task
     >>> thresh = Nipype1Task(fsl.Threshold())
