From c2fe15d1284753e1e5054227595d2e81e376bb56 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Sat, 9 Sep 2023 10:59:52 -0700 Subject: [PATCH] add current path for wyvern run (#54) * add current path for wyvern run * version bump --- pyproject.toml | 2 +- wyvern/cli/commands.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 1b73e27..b7a78c1 100644 --- 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 "] readme = "README.md" diff --git a/wyvern/cli/commands.py b/wyvern/cli/commands.py index b4ef561..754d9ad 100644 --- 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: