Skip to content

Commit ba1e449

Browse files
saurabh3949laurenyu
authored andcommitted
Init session in transform
1 parent 20f4353 commit ba1e449

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/sagemaker/model.py

+6
Original file line numberDiff line numberDiff line change
@@ -514,6 +514,12 @@ def transformer(
514514
volume_kms_key (str): Optional. KMS key ID for encrypting the volume
515515
attached to the ML compute instance (default: None).
516516
"""
517+
if not self.sagemaker_session:
518+
if instance_type in ("local", "local_gpu"):
519+
self.sagemaker_session = local.LocalSession()
520+
else:
521+
self.sagemaker_session = session.Session()
522+
517523
self._create_sagemaker_model(instance_type, tags=tags)
518524
if self.enable_network_isolation():
519525
env = None

0 commit comments

Comments
 (0)