@@ -55,19 +55,24 @@ def modulevolume():
55
55
56
56
@pytest .fixture (scope = 'module' , autouse = True )
57
57
def container (request , docker_base_name , tag ):
58
+ module_dir = os .path .join (resource_path , 'module' )
59
+ model_dir = os .path .join (resource_path , 'models' )
60
+
58
61
test_name = 'sagemaker-sklearn-serving-test'
59
62
try :
60
63
command = (
61
64
'docker run --name {} -p 8080:8080'
62
- ' --mount type=volume,source=dynamic_endpoint_model_volume,target=/opt/ml/model,readonly'
63
- ' --mount type=volume,source=dynamic_endpoint_module_volume,target=/user_module,readonly'
65
+ # ' --mount type=volume,source=dynamic_endpoint_model_volume,target=/opt/ml/model,readonly'
66
+ # ' --mount type=volume,source=dynamic_endpoint_module_volume,target=/user_module,readonly'
67
+ ' -v {}:/opt/ml/model'
68
+ ' -v {}:/user_module'
64
69
' -e SAGEMAKER_BIND_TO_PORT=8080'
65
70
' -e SAGEMAKER_SAFE_PORT_RANGE=9000-9999'
66
71
' -e SAGEMAKER_MULTI_MODEL=true'
67
72
' -e SAGEMAKER_PROGRAM={}'
68
73
' -e SAGEMAKER_SUBMIT_DIRECTORY={}'
69
74
' {}:{} serve'
70
- ).format (test_name , 'script.py' , "/user_module/user_code.tar.gz" , docker_base_name , tag )
75
+ ).format (test_name , model_dir , module_dir , 'script.py' , "/user_module/user_code.tar.gz" , docker_base_name , tag )
71
76
72
77
proc = subprocess .Popen (command .split (), stdout = sys .stdout , stderr = subprocess .STDOUT )
73
78
0 commit comments