no test for jit compile in torch

innat · Mar 25, 2024 · a626b1f · a626b1f
1 parent e3e02dc
commit a626b1f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/keras_cv/models/classification/video_classifier_test.py b/keras_cv/models/classification/video_classifier_test.py
@@ -51,6 +51,9 @@ def test_valid_call(self):
     )
     @pytest.mark.large  # Fit is slow, so mark these large.
     def test_classifier_fit(self, jit_compile):
+        if jit_compile and keras.backend.backend() == "torch":
+            self.skipTest("TODO: Torch Backend `jit_compile` fails on GPU.")
+            self.supports_jit = False
         model = VideoClassifier(
             backbone=VideoSwinBackbone(
                 input_shape=(8, 224, 224, 3), include_rescaling=True