Skip to content

Commit 72c096f

Browse files
committed
feat: update scikit-learn version to 1.4.2
1 parent b985f61 commit 72c096f

File tree

11 files changed

+12
-12
lines changed

11 files changed

+12
-12
lines changed

docker/1.2-1-1/base/Dockerfile.cpu renamed to docker/1.4-2/base/Dockerfile.cpu

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
2525
FROM ubuntu:${UBUNTU_VERSION}@sha256:${UBUNTU_IMAGE_DIGEST}
2626

2727
ARG MINICONDA_VERSION=24.7.1
28-
ARG CONDA_CHECKSUM=2006a61abc8b4fd04de5eb92620e1f72bada713cc84b5b4899463095e1210556
29-
ARG CONDA_PY_VERSION=39
28+
ARG CONDA_CHECKSUM=684cda724bc37e3bbbb342e440fc4cac515c92e91a489eb4359feca35382894b
29+
ARG CONDA_PY_VERSION=310
3030
ARG CONDA_PKG_VERSION=24.7.1
3131
ARG PYTHON_VERSION=3.10
3232
ARG PYARROW_VERSION=14.0.1
@@ -195,4 +195,4 @@ ENV PYTHONDONTWRITEBYTECODE=1 PYTHONUNBUFFERED=1 PYTHONIOENCODING=UTF-8 LANG=C.U
195195
# Install Scikit-Learn
196196
# Scikit-learn 0.20 was the last version to support Python 2.7 and Python 3.4.
197197
# Scikit-learn now requires Python 3.6 or newer.
198-
RUN python3 -m pip install --no-cache -I scikit-learn==1.2.1
198+
RUN python3 -m pip install --no-cache -I scikit-learn==1.4.2

docker/1.2-1-1/extension/Dockerfile.cpu renamed to docker/1.4-2/extension/Dockerfile.cpu

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
FROM preprod-sklearn:1.2-1
22

3-
RUN pip freeze | grep -q 'scikit-learn==1.2.1'; \
3+
RUN pip freeze | grep -q 'scikit-learn==1.4.2'; \
44
if [ $? -eq 0 ]; \
5-
then echo 'scikit-learn version 1.2.1 requirement met'; \
6-
else echo 'ERROR: Expected scikit-learn version is 1.2.1, check base images for scikit-learn version' && \
5+
then echo 'scikit-learn version 1.4.2 requirement met'; \
6+
else echo 'ERROR: Expected scikit-learn version is 1.4.2, check base images for scikit-learn version' && \
77
exit 1; fi
88

99
RUN pip install --upgrade --no-cache --no-deps sagemaker-scikit-learn-extension==2.5.0
File renamed without changes.

docker/1.2-1-1/final/Dockerfile.cpu renamed to docker/1.4-2/final/Dockerfile.cpu

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,12 @@
1-
FROM sklearn-base:1.2-1-1
2-
ENV SAGEMAKER_SKLEARN_VERSION 1.2-1-1
1+
FROM sklearn-base:1.4-2
2+
ENV SAGEMAKER_SKLEARN_VERSION 1.4-2
33
ENV PIP_ROOT_USER_ACTION=ignore
44

55
LABEL com.amazonaws.sagemaker.capabilities.accept-bind-to-port=true
66

77
COPY requirements.txt /requirements.txt
88
RUN python -m pip install -r /requirements.txt && \
9+
python -m pip install --force-reinstall numpy scipy pandas==1.4.2 scikit-learn==1.4.2 && \
910
rm /requirements.txt
1011

1112
COPY dist/sagemaker_sklearn_container-2.0-py3-none-any.whl /sagemaker_sklearn_container-2.0-py3-none-any.whl

requirements.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,7 @@ retrying==1.3.3
1414
sagemaker-containers==2.8.6.post2
1515
sagemaker-inference==1.2.0
1616
sagemaker-training==4.8.0
17-
scikit-learn==1.2.1
18-
scipy==1.9.3
17+
scikit-learn==1.4.2
1918
urllib3==1.26.17
2019
six==1.15.0
2120
jinja2==3.0.3

test/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def pytest_addoption(parser):
4343
parser.addoption('--install-container-support', '-C', action='store_true')
4444
parser.addoption('--docker-base-name', default='sk-learn')
4545
parser.addoption('--region', default='us-west-2')
46-
parser.addoption('--framework-version', default='1.2.1')
46+
parser.addoption('--framework-version', default='1.4.2')
4747
parser.addoption('--py-version', choices=['2', '3'], default=str(sys.version_info.major))
4848
parser.addoption('--processor', choices=['cpu'], default='cpu')
4949
# If not specified, will default to {framework-version}-{processor}-py{py-version}

0 commit comments

Comments
 (0)