update: video classifier test input shape

innat · Mar 6, 2024 · 146f32f · 146f32f
1 parent e2f5056
commit 146f32f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/keras_cv/models/classification/video_classifier_test.py b/keras_cv/models/classification/video_classifier_test.py
@@ -70,7 +70,9 @@ def test_classifier_fit(self, jit_compile):
     )
     def test_pooling_arg_call(self, pooling):
         model = VideoClassifier(
-            backbone=VideoSwinBackbone(include_rescaling=False),
+            backbone=VideoSwinBackbone(
+                input_shape=(8, 224, 224, 3), include_rescaling=True
+            ),
             num_classes=10,
             pooling=pooling,
         )