Skip to content

Commit ef16851

Browse files
committed
Initialize framework and version in post_init
1 parent fb0d789 commit ef16851

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

sagemaker-serve/src/sagemaker/serve/model_builder.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -489,6 +489,11 @@ def _initialize_defaults(self) -> None:
489489
self.container_config = "host"
490490
self.inference_recommender_job_results = None
491491
self.container_log_level = logging.INFO
492+
493+
if not hasattr(self, 'framework') or self.framework is None:
494+
self.framework = None
495+
if not hasattr(self, 'framework_version') or self.framework_version is None:
496+
self.framework_version = None
492497

493498
def _fetch_default_instance_type_for_custom_model(self) -> str:
494499
hosting_configs = self._fetch_hosting_configs_for_custom_model()

0 commit comments

Comments
 (0)