Fix for last index of module path (#2881)

* Update the generated component classes table. * Added back the MLflowReceiver. * Change to match the last module_path from source_file. --------- Co-authored-by: Chester Chen <[email protected]>
NVIDIA · Aug 30, 2024 · 9bb5879 · 9bb5879
1 parent 4279625
commit 9bb5879
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nvflare/job_config/fed_job_config.py b/nvflare/job_config/fed_job_config.py
@@ -203,7 +203,7 @@ def _get_custom_file(self, custom_dir, module, source_file):
             if package not in FL_PACKAGES and module not in self.custom_modules:
                 module_path = module.replace(".", os.sep)
                 if module_path in source_file:
-                    index = source_file.index(module_path)
+                    index = source_file.rindex(module_path)
                     dest = source_file[index:]
 
                     self.custom_modules.append(module)