Fix sagemaker-core unit test and sagemaker-serve integ test (#5509)

* Fix sagemaker-core unit test and sagemaker-serve integ test failure due to new djl images and pandas mock leak

Author: mollyheamazon

sagemaker-core/tests/unit/image_uris/test_djl.py

@@ -86,6 +86,12 @@ def _test_djl_uris(account, region, version, tag, djl_framework):
 # Known missing framework:version:region combinations that don't exist in ECR
 KNOWN_MISSING_COMBINATIONS = {
     "djl-lmi": {
+        "0.36.0-lmi18.0.0-cu128": {"ap-east-2"},
+        "0.35.0-lmi17.0.0-cu128": {"ap-east-2"},
+        "0.34.0-lmi16.0.0-cu128": {"ap-east-2"},
+        "0.33.0-lmi15.0.0-cu128": {"ap-east-2"},
+        "0.32.0-lmi14.0.0-cu126": {"ap-east-2"},
+        "0.31.0-lmi13.0.0-cu124": {"ap-east-2"},
         "0.30.0-lmi12.0.0-cu124": {"ap-east-2"},
         "0.29.0-lmi11.0.0-cu124": {"ap-east-2"},
         "0.28.0-lmi10.0.0-cu124": {"ap-east-2"},
@@ -205,6 +211,8 @@ def _test_djl_uris(account, region, version, tag, djl_framework):
         },
     },
     "djl-tensorrtllm": {
+        "0.33.0-tensorrtllm0.21.0-cu128": {"ap-east-2"},
+        "0.32.0-tensorrtllm0.12.0-cu125": {"ap-east-2"},
         "0.30.0-tensorrtllm0.12.0-cu125": {"ap-east-2"},
         "0.29.0-tensorrtllm0.11.0-cu124": {"ap-east-2"},
         "0.28.0-tensorrtllm0.9.0-cu122": {"ap-east-2"},

sagemaker-core/tests/unit/test_analytics.py

@@ -17,12 +17,8 @@
 import pytest
 from unittest.mock import Mock, patch, MagicMock, mock_open
 from collections import OrderedDict
-
-# Mock pandas before importing analytics
 import sys
 
-sys.modules["pandas"] = MagicMock()
-
 from sagemaker.core.analytics import (
     AnalyticsMetricsBase,
     HyperparameterTuningJobAnalytics,
@@ -33,6 +29,19 @@
 )
 
 
+@pytest.fixture(autouse=True)
+def mock_pandas_for_analytics():
+    """Mock pandas for analytics tests only, with proper cleanup."""
+    original_pandas = sys.modules.get("pandas")
+    sys.modules["pandas"] = MagicMock()
+    yield
+    # Restore original pandas after each test
+    if original_pandas is not None:
+        sys.modules["pandas"] = original_pandas
+    elif "pandas" in sys.modules:
+        del sys.modules["pandas"]
+
+
 class TestAnalyticsMetricsBase:
     """Test AnalyticsMetricsBase abstract class."""
 

sagemaker-serve/tests/integ/test_huggingface_integration.py

@@ -25,7 +25,7 @@
 logger = logging.getLogger(__name__)
 
 # Configuration - easily customizable
-MODEL_ID = "t5-small"  # Small text generation model
+MODEL_ID = "gpt2"  # Small causal language model compatible with DJL 0.36.0 vLLM
 MODEL_NAME_PREFIX = "hf-test-model"
 ENDPOINT_NAME_PREFIX = "hf-test-endpoint"