Skip to content

Commit 7a8cd34

Browse files
author
pintaoz
committed
Fix codestyle
1 parent 12456b1 commit 7a8cd34

File tree

3 files changed

+2
-2
lines changed

3 files changed

+2
-2
lines changed

src/sagemaker/pipeline.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -563,7 +563,6 @@ def delete_model(self):
563563

564564
self.sagemaker_session.delete_model(self.name)
565565

566-
567566
def _init_sagemaker_session_if_does_not_exist(self, instance_type=None):
568567
"""Set ``self.sagemaker_session`` to ``LocalSession`` or ``Session`` if it's not already.
569568

src/sagemaker/test.py

Lines changed: 0 additions & 1 deletion
This file was deleted.

src/sagemaker/workflow/steps.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -645,6 +645,7 @@ def arguments(self) -> RequestType:
645645
request_dict = self.step_args
646646
else:
647647
if isinstance(self.model, PipelineModel):
648+
self.model._init_sagemaker_session_if_does_not_exist()
648649
request_dict = self.model.sagemaker_session._create_model_request(
649650
name="",
650651
role=self.model.role,
@@ -653,6 +654,7 @@ def arguments(self) -> RequestType:
653654
enable_network_isolation=self.model.enable_network_isolation,
654655
)
655656
else:
657+
self.model._init_sagemaker_session_if_does_not_exist()
656658
request_dict = self.model.sagemaker_session._create_model_request(
657659
name="",
658660
role=self.model.role,

0 commit comments

Comments
 (0)