From b250f8bb2c0421d23f7b78aa6c966724343e7b44 Mon Sep 17 00:00:00 2001 From: jiapinw Date: Wed, 21 Feb 2024 15:50:14 -0800 Subject: [PATCH] Fix telemetry image option logic --- src/sagemaker/serve/builder/model_builder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sagemaker/serve/builder/model_builder.py b/src/sagemaker/serve/builder/model_builder.py index c7f2dc5633..b2f51e3a94 100644 --- a/src/sagemaker/serve/builder/model_builder.py +++ b/src/sagemaker/serve/builder/model_builder.py @@ -605,7 +605,7 @@ def build( self.serve_settings = self._get_serve_setting() - self._is_custom_image_uri = self.image_uri is None + self._is_custom_image_uri = self.image_uri is not None if isinstance(self.model, str): if self._is_jumpstart_model_id():