File tree Expand file tree Collapse file tree 2 files changed +5
-1
lines changed Expand file tree Collapse file tree 2 files changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -598,6 +598,7 @@ def register(
598598 source_uri = source_uri ,
599599 model_card = model_card ,
600600 )
601+ print ("Model package args are:" , ** model_pkg_args )
601602 model_package = self .sagemaker_session .create_model_package_from_containers (
602603 ** model_pkg_args
603604 )
@@ -977,6 +978,9 @@ def _create_sagemaker_model(
977978 enable_network_isolation = self ._enable_network_isolation ,
978979 tags = format_tags (tags ),
979980 )
981+ print ("ruiliann test create_model_args:" , create_model_args ) # Print create_model_args
982+
983+
980984 self .sagemaker_session .create_model (** create_model_args )
981985
982986 def _get_model_uri (self ):
Original file line number Diff line number Diff line change @@ -343,7 +343,7 @@ def _auto_detect_container(self):
343343 # Auto detect the container image uri
344344 if self .image_uri :
345345 logger .info (
346- "Skipping auto detection as the image uri is provided %s" ,
346+ "Test Skipping auto detection as the image uri is provided %s" ,
347347 self .image_uri ,
348348 )
349349 return
You can’t perform that action at this time.
0 commit comments