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(
598
598
source_uri = source_uri ,
599
599
model_card = model_card ,
600
600
)
601
+ print ("Model package args are:" , ** model_pkg_args )
601
602
model_package = self .sagemaker_session .create_model_package_from_containers (
602
603
** model_pkg_args
603
604
)
@@ -977,6 +978,9 @@ def _create_sagemaker_model(
977
978
enable_network_isolation = self ._enable_network_isolation ,
978
979
tags = format_tags (tags ),
979
980
)
981
+ print ("ruiliann test create_model_args:" , create_model_args ) # Print create_model_args
982
+
983
+
980
984
self .sagemaker_session .create_model (** create_model_args )
981
985
982
986
def _get_model_uri (self ):
Original file line number Diff line number Diff line change @@ -343,7 +343,7 @@ def _auto_detect_container(self):
343
343
# Auto detect the container image uri
344
344
if self .image_uri :
345
345
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" ,
347
347
self .image_uri ,
348
348
)
349
349
return
You can’t perform that action at this time.
0 commit comments