tests/datafiles: replace pkg_resources with importlib.resources

Becksteinlab · Jun 17, 2024 · 0c4bafc · 0c4bafc
1 parent a00ee59
commit 0c4bafc
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/datafiles.py b/tests/datafiles.py
@@ -16,9 +16,14 @@
 """
 
 
+import sys
 import os.path
-from pkg_resources import resource_filename
+from importlib import resources
 
 
 def datafile(name):
-    return resource_filename(__name__, os.path.join("data", name))
+    return (
+        resources.files(sys.modules[__name__].__package__)
+        .joinpath("data")
+        .joinpath(name)
+    )