Skip to content

Commit 2daf130

Browse files
Nikhil Raverkarawsbmillare
Nikhil Raverkar
authored andcommitted
upgrading sklearn to 1.0.2
1 parent 398037d commit 2daf130

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

test/integration/test_multiple_model_endpoint.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
path = os.path.abspath(__file__)
2929
resource_path = os.path.join(os.path.dirname(path), '..', 'resources')
3030

31+
3132
@pytest.fixture(scope='session', autouse=True)
3233
def volume():
3334
try:
@@ -61,16 +62,16 @@ def container(request, docker_base_name, tag):
6162
command = (
6263
'docker run --name {} -p 8080:8080'
6364
#' --mount type=volume,source=dynamic_endpoint_model_volume,target=/opt/ml/model,readonly'
64-
' --mount type=volume,source=dynamic_endpoint_module_volume,target=/user_module,readonly'
65+
#' --mount type=volume,source=dynamic_endpoint_module_volume,target=/user_module,readonly'
6566
' -v {}:/opt/ml/model'
66-
' -v {}:/user_module2'
67+
' -v {}:/user_module'
6768
' -e SAGEMAKER_BIND_TO_PORT=8080'
6869
' -e SAGEMAKER_SAFE_PORT_RANGE=9000-9999'
6970
' -e SAGEMAKER_MULTI_MODEL=true'
7071
' -e SAGEMAKER_PROGRAM={}'
7172
' -e SAGEMAKER_SUBMIT_DIRECTORY={}'
7273
' {}:{} serve'
73-
).format(test_name, model_dir, module_dir, 'script.py', "/user_module2/user_code.tar.gz", docker_base_name, tag)
74+
).format(test_name, model_dir, module_dir, 'script.py', "/user_module/user_code.tar.gz", docker_base_name, tag)
7475

7576
proc = subprocess.Popen(command.split(), stdout=sys.stdout, stderr=subprocess.STDOUT)
7677

0 commit comments

Comments
 (0)