aws / aws/sagemaker-scikit-learn-container
Failure to use correct version of Sklearn
- Dominant language
- Python
- Stars
- 188
- Forks
- 121
- PR merge metrics
- No merged PRs in 30d
Description
When deploying a mme endpoint, the sklearn version does not seem to use the correct version. I specify the
```
IMAGE_NAME = amazonaws.com/sagemaker-scikit-learn:1.0-1-cpu-py3
sklearn_estimator = SKLearnModel(
source_dir= source_dir
model_data= model_data,
entry_point="inference.py", # all models in MultiDataModel need to have inference.py
role=ROLE,
image_uri=IMAGE_NAME,
env={
"SAGEMAKER_SUBMIT_DIRECTORY": "/opt/ml/model/"
# 'SAGEMAKER_PROGRAM':
},
sagemaker_session=sagemaker_session,
)
```
I would expect this to use `sklearn==1.0.1`, but when i check logs it seems to be using `sklearn==0.23.2`
Any ideas that would cause this?
Contributor guide
Assessment
This issue has not been assessed yet.