Skip to content

Commit ea83cc6

Browse files
matthewfollegotlaurenyu
authored andcommitted
Addressing Issue aws#1184 - create Session by default
1 parent ba1e449 commit ea83cc6

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")
@@ -108,7 +109,7 @@ def __init__(
108109
self.env = env or {}
109110
self.name = name
110111
self.vpc_config = vpc_config
111-
self.sagemaker_session = sagemaker_session
112+
self.sagemaker_session = sagemaker_session or Session()
112113
self._model_name = None
113114
self.endpoint_name = None
114115
self._is_compiled_model = False

0 commit comments

Comments
 (0)