batch mismatch hotfix (#2112)

pkgoogle · web-flow · commit 5551f54aea4b · 2025-02-25T14:50:24.000-08:00
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

Original file line number	Diff line number	Diff line change
`@@ -89,7 +89,7 @@ def test_all_presets(self):`
`89`	`89`	`cls=MobileNetImageClassifier,`
`90`	`90`	`preset=preset,`
`91`	`91`	`input_data=self.images,`
`92`		`- expected_output_shape=(1, 1000),`
	`92`	`+ expected_output_shape=(2, 1000),`
`93`	`93`	`)`
`94`	`94`
`95`	`95`	`@pytest.mark.large`