add current path for wyvern run (#54)

* add current path for wyvern run * version bump
Skyvern-AI · Sep 9, 2023 · c2fe15d · c2fe15d
1 parent cefd422
commit c2fe15d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "wyvern-ai"
-version = "0.0.15"
+version = "0.0.16"
 description = ""
 authors = ["Wyvern AI <[email protected]>"]
 readme = "README.md"

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: