@@ -35,7 +35,7 @@ def create_docker(name: str, is_gpu: bool) -> str:
35
35
rand = str (uuid .uuid4 ())[:5 ]
36
36
tag = f"{ name } :{ rand } "
37
37
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 )
39
39
):
40
40
subprocess .run (["docker" , "build" , "." , "-t" , tag ])
41
41
return tag
@@ -84,12 +84,12 @@ def get_repo_name(model_id: str, dataset_name: str) -> str:
84
84
85
85
def show (args ):
86
86
directory = os .path .join (
87
- os .path .dirname (os .path .dirname (__file__ )), "docker_images"
87
+ os .path .dirname (os .path .normpath (__file__ )), "docker_images"
88
88
)
89
89
for framework in sorted (os .listdir (directory )):
90
90
print (f"{ framework } " )
91
91
local_path = os .path .join (
92
- os .path .dirname (os .path .dirname (__file__ )),
92
+ os .path .dirname (os .path .normpath (__file__ )),
93
93
"docker_images" ,
94
94
framework ,
95
95
"app" ,
@@ -151,7 +151,7 @@ def start(args):
151
151
model_id , task , framework = resolve_task_framework (args )
152
152
153
153
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
155
155
)
156
156
sys .path .append (local_path )
157
157
os .environ ["MODEL_ID" ] = model_id
0 commit comments