Integ test updates

samruds · samruds · commit 312d837ec50c · 2024-05-17T07:32:48.000-07:00
diff --git a/src/sagemaker/serve/builder/tei_builder.py b/src/sagemaker/serve/builder/tei_builder.py
@@ -94,7 +94,8 @@ def _create_tei_model(self,  **kwargs) -> Type[Model]:
             self.image_uri = image_uris.retrieve(
                 "huggingface-tei",
                 image_scope="inference",
-                instance_type=kwargs.get("instance_type")
+                instance_type=kwargs.get("instance_type"),
+                region=self.sagemaker_session.boto_region_name
             )
 
         pysdk_model = HuggingFaceModel(

Original file line number	Diff line number	Diff line change
`@@ -94,7 +94,8 @@ def _create_tei_model(self, **kwargs) -> Type[Model]:`
`94`	`94`	`self.image_uri = image_uris.retrieve(`
`95`	`95`	`"huggingface-tei",`
`96`	`96`	`image_scope="inference",`
`97`		`- instance_type=kwargs.get("instance_type")`
	`97`	`+ instance_type=kwargs.get("instance_type"),`
	`98`	`+ region=self.sagemaker_session.boto_region_name`
`98`	`99`	`)`
`99`	`100`
`100`	`101`	`pysdk_model = HuggingFaceModel(`