File tree 3 files changed +3
-2
lines changed
3 files changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -563,7 +563,7 @@ def delete_model(self):
563
563
564
564
self .sagemaker_session .delete_model (self .name )
565
565
566
-
566
+
567
567
def _init_sagemaker_session_if_does_not_exist (self , instance_type = None ):
568
568
"""Set ``self.sagemaker_session`` to ``LocalSession`` or ``Session`` if it's not already.
569
569
Load Diff This file was deleted.
Original file line number Diff line number Diff line change @@ -645,6 +645,7 @@ def arguments(self) -> RequestType:
645
645
request_dict = self .step_args
646
646
else :
647
647
if isinstance (self .model , PipelineModel ):
648
+ self .model ._init_sagemaker_session_if_does_not_exist ()
648
649
request_dict = self .model .sagemaker_session ._create_model_request (
649
650
name = "" ,
650
651
role = self .model .role ,
@@ -653,6 +654,7 @@ def arguments(self) -> RequestType:
653
654
enable_network_isolation = self .model .enable_network_isolation ,
654
655
)
655
656
else :
657
+ self .model ._init_sagemaker_session_if_does_not_exist ()
656
658
request_dict = self .model .sagemaker_session ._create_model_request (
657
659
name = "" ,
658
660
role = self .model .role ,
You can’t perform that action at this time.
0 commit comments