Update GPU info in test_hardware_detector

ruiliann666 · web-flow · commit 31af95a1f5c7 · 2025-04-03T15:42:35.000-07:00
diff --git a/tests/integ/sagemaker/serve/utils/test_hardware_detector.py b/tests/integ/sagemaker/serve/utils/test_hardware_detector.py
@@ -19,7 +19,7 @@
 REGION = "us-west-2"
 VALID_INSTANCE_TYPE = "ml.g5.48xlarge"
 INVALID_INSTANCE_TYPE = "fl.c5.57xxlarge"
-EXPECTED_INSTANCE_GPU_INFO = (8, 196608)
+EXPECTED_INSTANCE_GPU_INFO = (8, 183104)
 
 
 def test_get_gpu_info_success(sagemaker_session):
