aws / aws/sagemaker-scikit-learn-container

Failure to use correct version of Sklearn

Open
#170 0 comments 0 reactions 0 assignees View on GitHub
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

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.