diff --git a/keras/metrics/iou_metrics_test.py b/keras/metrics/iou_metrics_test.py index 92fb1c57404..7f5941454de 100644 --- 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(