Fix CI

keras-team · Nov 29, 2023 · 3379de9 · 3379de9
1 parent 5b9f1b7
commit 3379de9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/keras/metrics/iou_metrics_test.py b/keras/metrics/iou_metrics_test.py
@@ -1,4 +1,5 @@
 import numpy as np
+import pytest
 
 from keras import layers
 from keras import models
@@ -95,6 +96,7 @@ def test_zero_and_non_zero_entries(self):
         expected_result = (1 / (1 + 1 - 1)) / 1
         self.assertAllClose(result, expected_result, atol=1e-3)
 
+    @pytest.mark.requires_trainable_backend
     def test_compilation(self):
         m_obj = metrics.MeanIoU(num_classes=2)
         model = models.Sequential(