Try alternative implementation of importlib file-based module loader

wbond · Aug 17, 2023 · 2f14b21 · 2f14b21
1 parent 32b67e3
commit 2f14b21
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -47,8 +47,10 @@ def _import_from(mod, path, mod_dir=None):
  mod_info = imp.find_module(mod_dir, [path])
  return imp.load_module(mod, *mod_info)
  else:
- mod_info = importlib.machinery.PathFinder().find_spec(mod_dir, [path])
- return importlib.import_module(mod, *mod_info)
+ spec = importlib.machinery.PathFinder().find_spec(mod_dir, [path])
+ module = importlib.util.module_from_spec(spec)
+ sys.modules[mod] = module
+ spec.loader.exec_module(module)
  except ImportError:
  return None