|
28 | 28 | path = os.path.abspath(__file__)
|
29 | 29 | resource_path = os.path.join(os.path.dirname(path), '..', 'resources')
|
30 | 30 |
|
31 |
| - |
32 | 31 | @pytest.fixture(scope='session', autouse=True)
|
33 | 32 | def volume():
|
34 | 33 | try:
|
@@ -56,18 +55,22 @@ def modulevolume():
|
56 | 55 | @pytest.fixture(scope='module', autouse=True)
|
57 | 56 | def container(request, docker_base_name, tag):
|
58 | 57 | test_name = 'sagemaker-sklearn-serving-test'
|
| 58 | + module_dir = os.path.join(resource_path, 'module') |
| 59 | + model_dir = os.path.join(resource_path, 'models') |
59 | 60 | try:
|
60 | 61 | command = (
|
61 | 62 | '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_model_volume,target=/opt/ml/model,readonly' |
63 | 64 | ' --mount type=volume,source=dynamic_endpoint_module_volume,target=/user_module,readonly'
|
| 65 | + ' -v {}:/opt/ml/model' |
| 66 | + ' -v {}:/user_module2' |
64 | 67 | ' -e SAGEMAKER_BIND_TO_PORT=8080'
|
65 | 68 | ' -e SAGEMAKER_SAFE_PORT_RANGE=9000-9999'
|
66 | 69 | ' -e SAGEMAKER_MULTI_MODEL=true'
|
67 | 70 | ' -e SAGEMAKER_PROGRAM={}'
|
68 | 71 | ' -e SAGEMAKER_SUBMIT_DIRECTORY={}'
|
69 | 72 | ' {}:{} serve'
|
70 |
| - ).format(test_name, 'script.py', "/user_module/user_code.tar.gz", docker_base_name, tag) |
| 73 | + ).format(test_name, model_dir, module_dir, 'script.py', "/user_module2/user_code.tar.gz", docker_base_name, tag) |
71 | 74 |
|
72 | 75 | proc = subprocess.Popen(command.split(), stdout=sys.stdout, stderr=subprocess.STDOUT)
|
73 | 76 |
|
|
0 commit comments