Skip to content

Commit 30169b3

Browse files
author
Jonathan Makunga
committed
Fix tests
1 parent 6a5504b commit 30169b3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/serve/utils/telemetry_logger.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def wrapper(self, *args, **kwargs):
7272
if self.model_server == ModelServer.DJL_SERVING or self.model_server == ModelServer.TGI:
7373
extra += f"&x-modelName={self.model}"
7474

75-
if self.sagemaker_session.endpoint_arn:
75+
if self.sagemaker_session and self.sagemaker_session.endpoint_arn:
7676
extra += f"&x-endpointArn={self.sagemaker_session.endpoint_arn}"
7777

7878
start_timer = perf_counter()

0 commit comments

Comments
 (0)