[xy] Fix no module error when using custom scripts from the project (m…

…age-ai#676) * [xy] Print sys path. * [xy] Add project path to sys path.
Aliang-CN · Jul 19, 2022 · 1bdf3c0 · 1bdf3c0
1 parent 637aa2a
commit 1bdf3c0
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/mage_ai/__init__.py b/mage_ai/__init__.py
@@ -18,6 +18,7 @@
 import asyncio
 import logging
 import os
+import sys
 
 
 MAX_NUM_OF_ROWS = 100_000
@@ -120,8 +121,9 @@ def clean(
 # --------------- Data preparation methods --------------- #
 
 
-def run(pipeline_uuid: str, repo_path: str = None) -> None:
-    repo_path = os.getcwd() if repo_path is None else os.path.abspath(repo_path)
-    pipeline = Pipeline(pipeline_uuid, repo_path)
+def run(pipeline_uuid: str, project_path: str = None) -> None:
+    project_path = os.getcwd() if project_path is None else os.path.abspath(project_path)
+    sys.path.append(os.path.dirname(project_path))
+    pipeline = Pipeline(pipeline_uuid, project_path)
 
     asyncio.run(pipeline.execute(analyze_outputs=False, update_status=False))