Skip to content

Commit baeb6ed

Browse files
committed
fix missing attribute in telemetry ut
1 parent edcb885 commit baeb6ed

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

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

+1
Original file line numberDiff line numberDiff line change
@@ -162,6 +162,7 @@ def test_capture_telemetry_decorator_no_call_when_disabled(self, mock_send_telem
162162
mock_model_builder = ModelBuilderMock()
163163
mock_model_builder.serve_settings.telemetry_opt_out = True
164164
mock_model_builder.image_uri = MOCK_DJL_CONTAINER
165+
mock_model_builder._is_custom_image_uri = False
165166
mock_model_builder.model = MOCK_HUGGINGFACE_ID
166167
mock_model_builder.model_server = ModelServer.DJL_SERVING
167168

0 commit comments

Comments
 (0)