Skip to content

Commit 82329ce

Browse files
authored
feature: model.py uses sagemaker-core Session (#1706)
1 parent 2cc2681 commit 82329ce

File tree

1 file changed

+2
-1
lines changed
  • sagemaker-serve/src/sagemaker/serve

1 file changed

+2
-1
lines changed

sagemaker-serve/src/sagemaker/serve/model.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,9 @@
5454
from sagemaker.predictor import PredictorBase
5555
from sagemaker.predictor_async import AsyncPredictor
5656
from sagemaker.serverless import ServerlessInferenceConfig
57-
from sagemaker.session import (Session, get_add_model_package_inference_args,
57+
from sagemaker.session import (get_add_model_package_inference_args,
5858
get_update_model_package_inference_args)
59+
from sagemaker_core.helper.session_helper import Session
5960
from sagemaker.transformer import Transformer
6061
from sagemaker.utils import (Tags, _resolve_routing_config, _validate_new_tags,
6162
format_tags, remove_tag_with_key,

0 commit comments

Comments
 (0)