diff --git a/docker/0.23-1/base/Dockerfile.cpu b/docker/0.23-1/base/Dockerfile.cpu index 79572d4..42da75e 100644 --- a/docker/0.23-1/base/Dockerfile.cpu +++ b/docker/0.23-1/base/Dockerfile.cpu @@ -7,7 +7,7 @@ ARG MINICONDA_VERSION=4.8.3 ARG CONDA_PY_VERSION=37 ARG CONDA_PKG_VERSION=4.9.0 ARG PYTHON_VERSION=3.7.10 -ARG PYARROW_VERSION=0.16.0 +ARG PYARROW_VERSION=3.0.0 ARG MLIO_VERSION=0.6.0 # Install python and other scikit-learn runtime dependencies diff --git a/test/integration/test_multiple_model_endpoint.py b/test/integration/test_multiple_model_endpoint.py index 0ffa268..5ca0168 100644 --- a/test/integration/test_multiple_model_endpoint.py +++ b/test/integration/test_multiple_model_endpoint.py @@ -55,19 +55,24 @@ def modulevolume(): @pytest.fixture(scope='module', autouse=True) def container(request, docker_base_name, tag): + module_dir = os.path.join(resource_path, 'module') + model_dir = os.path.join(resource_path, 'models') + test_name = 'sagemaker-sklearn-serving-test' try: command = ( 'docker run --name {} -p 8080:8080' - ' --mount type=volume,source=dynamic_endpoint_model_volume,target=/opt/ml/model,readonly' - ' --mount type=volume,source=dynamic_endpoint_module_volume,target=/user_module,readonly' + # ' --mount type=volume,source=dynamic_endpoint_model_volume,target=/opt/ml/model,readonly' + # ' --mount type=volume,source=dynamic_endpoint_module_volume,target=/user_module,readonly' + ' -v {}:/opt/ml/model' + ' -v {}:/user_module' ' -e SAGEMAKER_BIND_TO_PORT=8080' ' -e SAGEMAKER_SAFE_PORT_RANGE=9000-9999' ' -e SAGEMAKER_MULTI_MODEL=true' ' -e SAGEMAKER_PROGRAM={}' ' -e SAGEMAKER_SUBMIT_DIRECTORY={}' ' {}:{} serve' - ).format(test_name, 'script.py', "/user_module/user_code.tar.gz", docker_base_name, tag) + ).format(test_name, model_dir, module_dir, 'script.py', "/user_module/user_code.tar.gz", docker_base_name, tag) proc = subprocess.Popen(command.split(), stdout=sys.stdout, stderr=subprocess.STDOUT) diff --git a/tox.ini b/tox.ini index d49cc64..93012b3 100644 --- a/tox.ini +++ b/tox.ini @@ -10,7 +10,7 @@ deps = -r{toxinidir}/requirements.txt -r{toxinidir}/test-requirements.txt conda_deps= - pyarrow=0.16.0 + pyarrow=3.0.0 mlio-py=0.5 conda_channels= conda-forge