batch mismatch hotfix (#2112)

keras-team · Feb 25, 2025 · 5551f54 · 5551f54
1 parent a5a16d8
commit 5551f54
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_hub/src/models/mobilenet/mobilenet_image_classifier_test.py b/keras_hub/src/models/mobilenet/mobilenet_image_classifier_test.py
@@ -89,7 +89,7 @@ def test_all_presets(self):
                 cls=MobileNetImageClassifier,
                 preset=preset,
                 input_data=self.images,
-                expected_output_shape=(1, 1000),
+                expected_output_shape=(2, 1000),
             )
 
     @pytest.mark.large