fix sys path

Author: pan-x-c

tests/utils/registry_test.py

@@ -1,13 +1,29 @@
 import unittest
 
-from trinity.common.workflows import WORKFLOWS, Workflow
+import ray
 
 
-class TestRegistry(unittest.TestCase):
-    def test_dynamic_import(self):
+class ImportUtils:
+    def run(self):
+        from trinity.common.workflows import WORKFLOWS, Workflow
+
         workflow_cls = WORKFLOWS.get("tests.utils.plugins.main.MainDummyWorkflow")
-        self.assertTrue(issubclass(workflow_cls, Workflow))
+        assert issubclass(workflow_cls, Workflow)
         workflow = workflow_cls(task=None, model=None)
         res = workflow.run()
-        self.assertEqual(res[0], 0)
-        self.assertEqual(res[1], "0")
+        assert res[0] == 0
+        assert res[1] == "0"
+
+
+class TestRegistry(unittest.TestCase):
+    def setUp(self):
+        ray.init(ignore_reinit_error=True)
+
+    def tearDown(self):
+        ray.shutdown()
+
+    def test_dynamic_import(self):
+        # test local import
+        ImportUtils().run()
+        # test remote import
+        ray.get(ray.remote(ImportUtils).remote().run.remote())

trinity/cli/launcher.py

@@ -248,12 +248,8 @@ def debug(
     load_plugins()
     config = load_config(config_path)
     config.check_and_update()
+    sys.path.insert(0, os.getcwd())
     config.ray_namespace = DEBUG_NAMESPACE
-    ray.init(
-        namespace=config.ray_namespace,
-        runtime_env={"env_vars": config.get_envs()},
-        ignore_reinit_error=True,
-    )
     from trinity.common.models import create_debug_inference_model
 
     if module == "inference_model":

trinity/utils/registry.py

@@ -1,3 +1,4 @@
+import sys
 import traceback
 from typing import Any, Type
 
@@ -145,6 +146,11 @@ def _dynamic_import(self, module_path: str, class_name: str) -> Type:
         """
         import importlib
 
-        module = importlib.import_module(module_path)
+        try:
+            module = importlib.import_module(module_path)
+        except ImportError:
+            self.logger.error(f"system path: {sys.path}")
+            self.logger.error(f"Cannot import module {module_path}")
+            raise ImportError(f"Cannot import module {module_path}")
         module_cls = getattr(module, class_name)
         return module_cls