Add Rich Logging to Model Builder (#1604)

nargokul · pintaoz-aws · commit c807e0d22469 · 2024-12-04T04:38:29.000-08:00
diff --git a/src/sagemaker/serve/builder/model_builder.py b/src/sagemaker/serve/builder/model_builder.py
@@ -117,8 +117,7 @@
 )
 from sagemaker.serve.validations.optimization import _validate_optimization_configuration
 from sagemaker.modules.train import ModelTrainer
-
-logger = logging.getLogger(__name__)
+from sagemaker.modules import logger
 
 # Any new server type should be added here
 supported_model_servers = {

Original file line number	Diff line number	Diff line change
`@@ -117,8 +117,7 @@`
`117`	`117`	`)`
`118`	`118`	`from sagemaker.serve.validations.optimization import _validate_optimization_configuration`
`119`	`119`	`from sagemaker.modules.train import ModelTrainer`
`120`		`-`
`121`		`-logger = logging.getLogger(__name__)`
	`120`	`+from sagemaker.modules import logger`
`122`	`121`
`123`	`122`	`# Any new server type should be added here`
`124`	`123`	`supported_model_servers = {`