Fix: JumpStartModelType proprietary register

selvask-aws · selvask-aws · commit 27b29de2951a · 2024-07-18T11:12:38.000+05:30
diff --git a/src/sagemaker/model.py b/src/sagemaker/model.py
@@ -551,6 +551,7 @@ def register(
             )
         if (
             model_package_group_name is not None
+            and self.model_type is not None
             and self.model_type is not JumpStartModelType.PROPRIETARY
         ):
             container_def = self.prepare_container_def(accept_eula=accept_eula)

Original file line number	Diff line number	Diff line change
`@@ -551,6 +551,7 @@ def register(`
`551`	`551`	`)`
`552`	`552`	`if (`
`553`	`553`	`model_package_group_name is not None`
	`554`	`+ and self.model_type is not None`
`554`	`555`	`and self.model_type is not JumpStartModelType.PROPRIETARY`
`555`	`556`	`):`
`556`	`557`	`container_def = self.prepare_container_def(accept_eula=accept_eula)`