From 672bebddfc9b33373a6c1bd9b077f3fcda96fcdf Mon Sep 17 00:00:00 2001 From: favilo Date: Fri, 21 Jun 2024 10:16:06 -0700 Subject: [PATCH] Fix rally-tracks-compat nox test --- esrally/track/loader.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/esrally/track/loader.py b/esrally/track/loader.py index 56a299d13..6c6f947e1 100644 --- 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)