Fix rally-tracks-compat nox test

elastic · Jun 21, 2024 · 672bebd · 672bebd
1 parent c8bae0a
commit 672bebd
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/esrally/track/loader.py b/esrally/track/loader.py
@@ -1226,10 +1226,11 @@ def load(self):
         # get dependent libraries installed in a prior step. ensure dir exists to make sure loading works correctly.
         os.makedirs(paths.libs(), exist_ok=True)
         sys.path.insert(0, paths.libs())
-        root_module = self.loader.load()
+        root_modules = self.loader.load()
         try:
             # every module needs to have a register() method
-            root_module.register(self)
+            for module in root_modules:
+                module.register(self)
         except BaseException:
             msg = "Could not register track plugin at [%s]" % self.loader.root_path
             logging.getLogger(__name__).exception(msg)