We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2cc2681 commit 82329ceCopy full SHA for 82329ce
sagemaker-serve/src/sagemaker/serve/model.py
@@ -54,8 +54,9 @@
54
from sagemaker.predictor import PredictorBase
55
from sagemaker.predictor_async import AsyncPredictor
56
from sagemaker.serverless import ServerlessInferenceConfig
57
-from sagemaker.session import (Session, get_add_model_package_inference_args,
+from sagemaker.session import (get_add_model_package_inference_args,
58
get_update_model_package_inference_args)
59
+from sagemaker_core.helper.session_helper import Session
60
from sagemaker.transformer import Transformer
61
from sagemaker.utils import (Tags, _resolve_routing_config, _validate_new_tags,
62
format_tags, remove_tag_with_key,
0 commit comments