Skip to content

Commit 9d5b059

Browse files
author
pintaoz
committed
Fix codestyle
1 parent 12456b1 commit 9d5b059

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

src/sagemaker/pipeline.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -563,7 +563,7 @@ def delete_model(self):
563563

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

566-
566+
567567
def _init_sagemaker_session_if_does_not_exist(self, instance_type=None):
568568
"""Set ``self.sagemaker_session`` to ``LocalSession`` or ``Session`` if it's not already.
569569

src/sagemaker/test.py

-1
This file was deleted.

src/sagemaker/workflow/steps.py

+2
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)