Skip to content

Commit 245f2ec

Browse files
committed
Capture create model args
1 parent ae6a9a2 commit 245f2ec

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

src/sagemaker/model.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff 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):

src/sagemaker/serve/builder/model_builder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)