Skip to content

Commit b0fd3dd

Browse files
authored
Improve manage by making it more agnostic to call method (#410)
* Improve manage by making it more agnostic to call method * Remove debugger.
1 parent 7a51428 commit b0fd3dd

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

manage.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def create_docker(name: str, is_gpu: bool) -> str:
3535
rand = str(uuid.uuid4())[:5]
3636
tag = f"{name}:{rand}"
3737
with cd(
38-
os.path.join(os.path.dirname(os.path.dirname(__file__)), "docker_images", name)
38+
os.path.join(os.path.dirname(os.path.normpath(__file__)), "docker_images", name)
3939
):
4040
subprocess.run(["docker", "build", ".", "-t", tag])
4141
return tag
@@ -84,12 +84,12 @@ def get_repo_name(model_id: str, dataset_name: str) -> str:
8484

8585
def show(args):
8686
directory = os.path.join(
87-
os.path.dirname(os.path.dirname(__file__)), "docker_images"
87+
os.path.dirname(os.path.normpath(__file__)), "docker_images"
8888
)
8989
for framework in sorted(os.listdir(directory)):
9090
print(f"{framework}")
9191
local_path = os.path.join(
92-
os.path.dirname(os.path.dirname(__file__)),
92+
os.path.dirname(os.path.normpath(__file__)),
9393
"docker_images",
9494
framework,
9595
"app",
@@ -151,7 +151,7 @@ def start(args):
151151
model_id, task, framework = resolve_task_framework(args)
152152

153153
local_path = os.path.join(
154-
os.path.dirname(os.path.dirname(__file__)), "docker_images", framework
154+
os.path.dirname(os.path.normpath(__file__)), "docker_images", framework
155155
)
156156
sys.path.append(local_path)
157157
os.environ["MODEL_ID"] = model_id

0 commit comments

Comments
 (0)