LIU-382: Demonstrate transition to importlib.resources

ICRAR · Jun 10, 2024 · 418ff91 · 418ff91
1 parent 0291fe3
commit 418ff91
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/daliuge-translator/test/dropmake/test_pg_gen.py b/daliuge-translator/test/dropmake/test_pg_gen.py
@@ -21,6 +21,11 @@
 
 import unittest
 
+try:
+    from importlib.resources import files, as_file
+except (ImportError, ModuleNotFoundError):
+    from importlib_resources import files
+
 import pkg_resources
 from dlg.common import CategoryType
 from dlg.dropmake.lg import LG
@@ -33,9 +38,7 @@
 
 
 def get_lg_fname(lg_name):
-    return pkg_resources.resource_filename(
-        __name__, "logical_graphs/{0}".format(lg_name)
-    )  # @UndefinedVariable
+    return str(files(__package__) / f"logical_graphs/{lg_name}")
 
 
 class TestPGGen(unittest.TestCase):