try to fix ut failure

Author: jiapinw

tests/unit/sagemaker/serve/utils/test_telemetry_logger.py

@@ -87,10 +87,10 @@ def test_capture_telemetry_decorator_djl_success(self, mock_send_telemetry):
             "&x-modelServer=4"
             "&x-imageTag=djl-inference:0.25.0-deepspeed0.11.0-cu118"
             f"&x-sdkVersion={SDK_VERSION}"
+            f"&x-defaultImageUsage={ImageUriOption.DEFAULT_IMAGE.value}"
             f"&x-modelName={MOCK_HUGGINGFACE_ID}"
             f"&x-endpointArn={MOCK_ENDPOINT_ARN}"
             f"&x-latency={latency}"
-            f"&x-defaultImageUsage={ImageUriOption.DEFAULT_IMAGE.value}"
         )
 
         mock_send_telemetry.assert_called_once_with(
@@ -117,10 +117,10 @@ def test_capture_telemetry_decorator_djl_success_with_custom_image(self, mock_se
             "&x-modelServer=4"
             "&x-imageTag=djl-inference:0.25.0-deepspeed0.11.0-cu118"
             f"&x-sdkVersion={SDK_VERSION}"
+            f"&x-defaultImageUsage={ImageUriOption.CUSTOM_1P_IMAGE.value}"
             f"&x-modelName={MOCK_HUGGINGFACE_ID}"
             f"&x-endpointArn={MOCK_ENDPOINT_ARN}"
             f"&x-latency={latency}"
-            f"&x-defaultImageUsage={ImageUriOption.CUSTOM_1P_IMAGE.value}"
         )
 
         mock_send_telemetry.assert_called_once_with(
@@ -147,10 +147,10 @@ def test_capture_telemetry_decorator_tgi_success(self, mock_send_telemetry):
             "&x-modelServer=6"
             "&x-imageTag=huggingface-pytorch-inference:2.0.0-transformers4.28.1-cpu-py310-ubuntu20.04"
             f"&x-sdkVersion={SDK_VERSION}"
+            f"&x-defaultImageUsage={ImageUriOption.DEFAULT_IMAGE.value}"
             f"&x-modelName={MOCK_HUGGINGFACE_ID}"
             f"&x-endpointArn={MOCK_ENDPOINT_ARN}"
             f"&x-latency={latency}"
-            f"&x-defaultImageUsage={ImageUriOption.DEFAULT_IMAGE.value}"
         )
 
         mock_send_telemetry.assert_called_once_with(
@@ -194,10 +194,10 @@ def test_capture_telemetry_decorator_handle_exception_success(self, mock_send_te
             "&x-modelServer=4"
             "&x-imageTag=djl-inference:0.25.0-deepspeed0.11.0-cu118"
             f"&x-sdkVersion={SDK_VERSION}"
+            f"&x-defaultImageUsage={ImageUriOption.DEFAULT_IMAGE.value}"
             f"&x-modelName={MOCK_HUGGINGFACE_ID}"
             f"&x-endpointArn={MOCK_ENDPOINT_ARN}"
             f"&x-latency={latency}"
-            f"&x-defaultImageUsage={ImageUriOption.DEFAULT_IMAGE.value}"
         )
 
         mock_send_telemetry.assert_called_once_with(