Skip to content

Commit db1ea55

Browse files
committed
feat: update skl version to 1.4.2
1 parent 72c096f commit db1ea55

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

docker/1.4-2/final/Dockerfile.cpu

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,15 @@ COPY requirements.txt /requirements.txt
88
RUN python -m pip install -r /requirements.txt && \
99
python -m pip install --force-reinstall numpy scipy pandas==1.4.2 scikit-learn==1.4.2 && \
1010
rm /requirements.txt
11+
12+
# Fix Python 3.10 compatibility for sagemaker-containers
13+
RUN python3 -c "import sys; sys.path.insert(0, '/miniconda3/lib/python3.10/site-packages'); \
14+
import sagemaker_containers._mapping as m; \
15+
import collections.abc; \
16+
setattr(collections, 'Mapping', collections.abc.Mapping); \
17+
exec(open('/miniconda3/lib/python3.10/site-packages/sagemaker_containers/_mapping.py').read().replace('collections.Mapping', 'collections.abc.Mapping'))" || \
18+
sed -i 's/collections\.Mapping/collections.abc.Mapping/g' /miniconda3/lib/python3.10/site-packages/sagemaker_containers/_mapping.py
19+
1120

1221
COPY dist/sagemaker_sklearn_container-2.0-py3-none-any.whl /sagemaker_sklearn_container-2.0-py3-none-any.whl
1322
RUN rm /miniconda3/lib/python3.10/site-packages/**/REQUESTED && \

0 commit comments

Comments
 (0)