From 5551f54aea4b226cbd9637b64c5a45c320ae0dbf Mon Sep 17 00:00:00 2001
From: pkgoogle <132095473+pkgoogle@users.noreply.github.com>
Date: Tue, 25 Feb 2025 22:50:24 +0000
Subject: [PATCH] batch mismatch hotfix (#2112)

---
 .../src/models/mobilenet/mobilenet_image_classifier_test.py     | 2 +-
 1 file changed, 1 insertion(+), 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
index 4cfefa1d68..a2011acafe 100644
--- 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