ModelDataUrl is made optional, applies to certain prebuilt containers and gen-ai dlami

pranavvm26 · pranavvm26 · commit 3d55ead27aa7 · 2025-01-21T23:42:02.000-05:00
diff --git a/src/sagemaker/local/entities.py b/src/sagemaker/local/entities.py
@@ -615,6 +615,10 @@ def serve(self):
         self.container = _SageMakerContainer(
             instance_type, instance_count, image, self.local_session
         )
+        
+        if "ModelDataUrl" not in  self.primary_container.keys():
+            self.primary_container["ModelDataUrl"] = None
+            
         self.container.serve(
             self.primary_container["ModelDataUrl"], self.primary_container["Environment"]
         )

Original file line number	Diff line number	Diff line change
`@@ -615,6 +615,10 @@ def serve(self):`
`615`	`615`	`self.container = _SageMakerContainer(`
`616`	`616`	`instance_type, instance_count, image, self.local_session`
`617`	`617`	`)`
	`618`	`+`
	`619`	`+ if "ModelDataUrl" not in self.primary_container.keys():`
	`620`	`+ self.primary_container["ModelDataUrl"] = None`
	`621`	`+`
`618`	`622`	`self.container.serve(`
`619`	`623`	`self.primary_container["ModelDataUrl"], self.primary_container["Environment"]`
`620`	`624`	`)`