fix scope of loading the registering the action based lib objects

OpenMined · Apr 9, 2024 · 049e1f0 · 049e1f0
1 parent c6c5cca
commit 049e1f0
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/packages/syft/src/syft/service/service.py b/packages/syft/src/syft/service/service.py
@@ -226,14 +226,15 @@ def register_lib_obj(lib_obj: CMPBase) -> None:
 
             LibConfigRegistry.register(lib_config)
 
-    # hacky, prevent circular imports
-    for lib_obj in action_execute_registry_libs.flatten():
-        # # for functions
-        # func_name = func.__name__
-        # # for classes
-        # func_name = path.split(".")[-1]
-        if isinstance(lib_obj, CMPFunction) or isinstance(lib_obj, CMPClass):
-            register_lib_obj(lib_obj)
+
+# hacky, prevent circular imports
+for lib_obj in action_execute_registry_libs.flatten():
+    # # for functions
+    # func_name = func.__name__
+    # # for classes
+    # func_name = path.split(".")[-1]
+    if isinstance(lib_obj, CMPFunction) or isinstance(lib_obj, CMPClass):
+        register_lib_obj(lib_obj)
 
 
 def deconstruct_param(param: inspect.Parameter) -> dict[str, Any]: