-
Notifications
You must be signed in to change notification settings - Fork 117
Open
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?
Metadata
Metadata
Assignees
Labels
No labels