Skip to content

Commit 398037d

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

File tree

3 files changed

+8
-5
lines changed

3 files changed

+8
-5
lines changed

docker/1.0-1/base/Dockerfile.cpu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@ ARG UBUNTU_IMAGE_DIGEST=98706f0f213dbd440021993a82d2f70451a73698315370ae8615cc46
33

44
FROM ubuntu:${UBUNTU_VERSION}@sha256:${UBUNTU_IMAGE_DIGEST}
55

6-
ARG MINICONDA_VERSION=4.8.3
7-
ARG CONDA_PY_VERSION=37
6+
ARG MINICONDA_VERSION=4.11.0
7+
ARG CONDA_PY_VERSION=39
88
ARG CONDA_PKG_VERSION=4.9.0
99
ARG PYTHON_VERSION=3.7.10
1010
ARG PYARROW_VERSION=0.16.0

test/integration/test_multiple_model_endpoint.py

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

31-
3231
@pytest.fixture(scope='session', autouse=True)
3332
def volume():
3433
try:
@@ -56,18 +55,22 @@ def modulevolume():
5655
@pytest.fixture(scope='module', autouse=True)
5756
def container(request, docker_base_name, tag):
5857
test_name = 'sagemaker-sklearn-serving-test'
58+
module_dir = os.path.join(resource_path, 'module')
59+
model_dir = os.path.join(resource_path, 'models')
5960
try:
6061
command = (
6162
'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'
6364
' --mount type=volume,source=dynamic_endpoint_module_volume,target=/user_module,readonly'
65+
' -v {}:/opt/ml/model'
66+
' -v {}:/user_module2'
6467
' -e SAGEMAKER_BIND_TO_PORT=8080'
6568
' -e SAGEMAKER_SAFE_PORT_RANGE=9000-9999'
6669
' -e SAGEMAKER_MULTI_MODEL=true'
6770
' -e SAGEMAKER_PROGRAM={}'
6871
' -e SAGEMAKER_SUBMIT_DIRECTORY={}'
6972
' {}:{} 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)
7174

7275
proc = subprocess.Popen(command.split(), stdout=sys.stdout, stderr=subprocess.STDOUT)
7376

-302 Bytes
Binary file not shown.

0 commit comments

Comments
 (0)