We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 20f4353 commit ba1e449Copy full SHA for ba1e449
src/sagemaker/model.py
@@ -514,6 +514,12 @@ def transformer(
514
volume_kms_key (str): Optional. KMS key ID for encrypting the volume
515
attached to the ML compute instance (default: None).
516
"""
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
+
523
self._create_sagemaker_model(instance_type, tags=tags)
524
if self.enable_network_isolation():
525
env = None
0 commit comments