forgot to call exec_module...

locustio · Nov 14, 2023 · b7952e2 · b7952e2
1 parent 2254e7c
commit b7952e2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/locust/util/load_locustfile.py b/locust/util/load_locustfile.py
@@ -56,7 +56,8 @@ def load_locustfile(path) -> Tuple[Optional[str], Dict[str, User], List[LoadTest
     loader = importlib.machinery.SourceFileLoader(os.path.splitext(locustfile)[0], path)
     spec = importlib.util.spec_from_loader(loader.name, loader)
     assert spec
-    imported = importlib.util.module_from_spec(spec)
+    imported_module = importlib.util.module_from_spec(spec)
+    loader.exec_module(imported_module)
 
     # Remove directory from path if we added it ourselves (just to be neat)
     if added_to_path: