Skip torch only test with other backends

keras-team · Aug 30, 2023 · 0b5f071 · 0b5f071
1 parent 18101a9
commit 0b5f071
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/keras_core/utils/torch_utils_test.py b/keras_core/utils/torch_utils_test.py
@@ -1,6 +1,8 @@
 import numpy as np
+import pytest
 import torch
 
+from keras_core import backend
 from keras_core import models
 from keras_core import testing
 from keras_core.utils.torch_utils import TorchModuleWrapper
@@ -15,6 +17,9 @@ def call(self, x):
         return self.fc(x)
 
 
+@pytest.mark.skipif(
+    backend.backend() != "torch", reason="Requires torch backend"
+)
 class TorchUtilsTest(testing.TestCase):
     def test_basic_usage(self):
         model = Classifier()