* extract if statement to the outside of the activate_data_module func.

Author: HYLcool

trinity/cli/launcher.py

@@ -95,20 +95,18 @@ def both(config: Config) -> None:
             logger.info("Eval step finished.")
 
 
-def activate_data_module(config: Config, config_path: str):
+def activate_data_module(config_path: str):
     """Check whether to activate data module and preprocess datasets."""
-    data_config = config.data
-    if data_config.dj_config_path or data_config.dj_process_desc:
-        from trinity.data.client import LOCAL_SERVER_URL, request
+    from trinity.data.client import LOCAL_SERVER_URL, request
 
-        logger.info("Activating data module...")
-        res = request(
-            url=LOCAL_SERVER_URL,
-            configPath=config_path,
-        )
-        if res["return_code"] != 0:
-            logger.error(f"Failed to activate data module: {res['return_msg']}.")
-            return
+    logger.info("Activating data module...")
+    res = request(
+        url=LOCAL_SERVER_URL,
+        configPath=config_path,
+    )
+    if res["return_code"] != 0:
+        logger.error(f"Failed to activate data module: {res['return_msg']}.")
+        return
 
 
 def main() -> None:
@@ -128,7 +126,9 @@ def main() -> None:
         config = load_config(args.config)
         config.check_and_update()
         # try to activate data module
-        activate_data_module(config, args.config)
+        data_config = config.data
+        if data_config.dj_config_path or data_config.dj_process_desc:
+            activate_data_module(args.config)
         ray.init()
         if config.mode == "explore":
             explore(config)