Update deprecated load_module (#21651)

2023-02-15 15:57:24 -05:00
parent 1567bef3b3
commit 9d1116e995
10 changed files with 39 additions and 10 deletions
--- a/tests/pipelines/test_pipelines_common.py
+++ b/tests/pipelines/test_pipelines_common.py
@@ -74,7 +74,9 @@ spec = importlib.util.spec_from_file_location(
    os.path.join(PATH_TO_TRANSFORMERS, "__init__.py"),
    submodule_search_locations=[PATH_TO_TRANSFORMERS],
 )
-transformers_module = spec.loader.load_module()
+transformers_module = importlib.util.module_from_spec(spec)
+spec.loader.exec_module(transformers_module)
+transformers_module = sys.modules["transformers"]


 class ANY: