fix: Incorrect preloading of the HardExampleMining module.

Signed-off-by: Yu Fan <[email protected]>
kubeedge · Sep 6, 2024 · 8b6c83b · 8b6c83b
1 parent f2352ce
commit 8b6c83b
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/core/testcasecontroller/algorithm/module/module.py b/core/testcasecontroller/algorithm/module/module.py
@@ -110,9 +110,14 @@ def get_module_instance(self, module_type):
         if self.url:
             try:
                 utils.load_module(self.url)
-                # pylint: disable=E1134
-                func = ClassFactory.get_cls(
-                    type_name=class_factory_type, t_cls_name=self.name)(**self.hyperparameters)
+
+                if class_factory_type == ClassType.HEM:
+                    func = {"method": self.name, "param":self.hyperparameters}
+                else:
+                    func = ClassFactory.get_cls(
+                        type_name=class_factory_type,
+                        t_cls_name=self.name
+                    )(**self.hyperparameters)
 
                 return func
             except Exception as err: