|
28 | 28 | path = os.path.abspath(__file__)
|
29 | 29 | resource_path = os.path.join(os.path.dirname(path), '..', 'resources')
|
30 | 30 |
|
| 31 | + |
31 | 32 | @pytest.fixture(scope='session', autouse=True)
|
32 | 33 | def volume():
|
33 | 34 | try:
|
@@ -61,16 +62,16 @@ def container(request, docker_base_name, tag):
|
61 | 62 | command = (
|
62 | 63 | 'docker run --name {} -p 8080:8080'
|
63 | 64 | #' --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' |
65 | 66 | ' -v {}:/opt/ml/model'
|
66 |
| - ' -v {}:/user_module2' |
| 67 | + ' -v {}:/user_module' |
67 | 68 | ' -e SAGEMAKER_BIND_TO_PORT=8080'
|
68 | 69 | ' -e SAGEMAKER_SAFE_PORT_RANGE=9000-9999'
|
69 | 70 | ' -e SAGEMAKER_MULTI_MODEL=true'
|
70 | 71 | ' -e SAGEMAKER_PROGRAM={}'
|
71 | 72 | ' -e SAGEMAKER_SUBMIT_DIRECTORY={}'
|
72 | 73 | ' {}:{} 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) |
74 | 75 |
|
75 | 76 | proc = subprocess.Popen(command.split(), stdout=sys.stdout, stderr=subprocess.STDOUT)
|
76 | 77 |
|
|
0 commit comments