add current path for wyvern run

Skyvern-AI · Sep 9, 2023 · 4d8bd28 · 4d8bd28
1 parent cefd422
commit 4d8bd28
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/wyvern/cli/commands.py b/wyvern/cli/commands.py
@@ -5,6 +5,7 @@
 import platform
 import shutil
 import subprocess
+import sys
 import zipfile
 from pathlib import Path
 from typing import Optional
@@ -175,6 +176,7 @@ def run(
     typer.echo("Running your ML application")
     # import the app from path
     try:
+        sys.path.append(".")
         module_path, app_name = path.split(":")
         module = importlib.import_module(module_path)
     except ImportError: