Skip to content

Commit 8e679f1

Browse files
authored
fix: pip installation path in 1.2-1-1 (#253)
1 parent 44a0cdc commit 8e679f1

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

docker/1.2-1-1/final/Dockerfile.cpu

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,10 +8,18 @@ COPY requirements.txt /requirements.txt
88
RUN python -m pip install -r /requirements.txt && \
99
rm /requirements.txt
1010

11+
# Fix Python 3.10 compatibility for sagemaker-containers
12+
RUN python3 -c "import sys; sys.path.insert(0, '/miniconda3/lib/python3.10/site-packages'); \
13+
import sagemaker_containers._mapping as m; \
14+
import collections.abc; \
15+
setattr(collections, 'Mapping', collections.abc.Mapping); \
16+
exec(open('/miniconda3/lib/python3.10/site-packages/sagemaker_containers/_mapping.py').read().replace('collections.Mapping', 'collections.abc.Mapping'))" || \
17+
sed -i 's/collections\.Mapping/collections.abc.Mapping/g' /miniconda3/lib/python3.10/site-packages/sagemaker_containers/_mapping.py
18+
1119
COPY dist/sagemaker_sklearn_container-2.0-py3-none-any.whl /sagemaker_sklearn_container-2.0-py3-none-any.whl
1220
RUN rm /miniconda3/lib/python3.10/site-packages/**/REQUESTED && \
1321
rm /miniconda3/lib/python3.10/site-packages/**/direct_url.json
14-
RUN pip install --no-cache /sagemaker_sklearn_container-2.0-py3-none-any.whl && \
22+
RUN python3 -m pip install --no-cache /sagemaker_sklearn_container-2.0-py3-none-any.whl && \
1523
rm /sagemaker_sklearn_container-2.0-py3-none-any.whl
1624

1725
ENV SAGEMAKER_TRAINING_MODULE sagemaker_sklearn_container.training:main

0 commit comments

Comments
 (0)