Skip to content

Commit 8cf40b1

Browse files
Addressing Issue aws#1184 - create Session by default
1 parent 5182be3 commit 8cf40b1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/sagemaker/model.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
import sagemaker
2121
from sagemaker import fw_utils, local, session, utils, git_utils
2222
from sagemaker.fw_utils import UploadedCode
23+
from sagemaker.session import Session
2324
from sagemaker.transformer import Transformer
2425

2526
LOGGER = logging.getLogger("sagemaker")
@@ -129,7 +130,7 @@ def __init__(
129130
self.env = env or {}
130131
self.name = name
131132
self.vpc_config = vpc_config
132-
self.sagemaker_session = sagemaker_session
133+
self.sagemaker_session = sagemaker_session or Session()
133134
self._model_name = None
134135
self.endpoint_name = None
135136
self._is_compiled_model = False

0 commit comments

Comments
 (0)