Skip to content

Commit f0c3c64

Browse files
jiapinwakrishna1995
authored andcommitted
fix pylint
1 parent aa21238 commit f0c3c64

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/sagemaker/serve/utils/telemetry_logger.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,12 +63,13 @@ def wrapper(self, *args, **kwargs):
6363
caught_ex = None
6464

6565
image_uri_tail = self.image_uri.split("/")[1]
66+
image_uri_option = _get_image_uri_option(self.image_uri, self._is_custom_image_uri)
6667
extra = (
6768
f"{func_name}"
6869
f"&x-modelServer={MODEL_SERVER_TO_CODE[str(self.model_server)]}"
6970
f"&x-imageTag={image_uri_tail}"
7071
f"&x-sdkVersion={SDK_VERSION}"
71-
f"&x-defaultImageUsage={_get_image_uri_option(self.image_uri, self._is_custom_image_uri)}"
72+
f"&x-defaultImageUsage={image_uri_option}"
7273
)
7374

7475
if self.model_server == ModelServer.DJL_SERVING or self.model_server == ModelServer.TGI:

0 commit comments

Comments
 (0)