From 0b5f071960117626cb68014ec4989d120012c55e Mon Sep 17 00:00:00 2001 From: Francois Chollet Date: Wed, 30 Aug 2023 15:52:08 -0700 Subject: [PATCH] Skip torch only test with other backends --- keras_core/utils/torch_utils_test.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/keras_core/utils/torch_utils_test.py b/keras_core/utils/torch_utils_test.py index 788ef9ecd..21d403c7a 100644 --- 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()