Skip to content

Commit

Permalink
test: skip panoptic quality for PT <1.12 (#2478)
Browse files Browse the repository at this point in the history
  • Loading branch information
Borda authored Mar 27, 2024
1 parent 086c611 commit 82e6abe
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tests/unittests/utilities/test_plot.py
Original file line number Diff line number Diff line change
Expand Up @@ -169,6 +169,7 @@
WordInfoPreserved,
)
from torchmetrics.utilities.imports import (
_TORCH_GREATER_EQUAL_1_12,
_TORCHAUDIO_GREATER_EQUAL_0_10,
)
from torchmetrics.utilities.plot import _get_col_row_split
Expand Down Expand Up @@ -341,6 +342,9 @@
_panoptic_input,
_panoptic_input,
id="panoptic quality",
marks=pytest.mark.skipif(
not _TORCH_GREATER_EQUAL_1_12, reason="Panoptic Quality metric requires PyTorch 1.12 or later"
),
),
pytest.param(BinaryAveragePrecision, _rand_input, _binary_randint_input, id="binary average precision"),
pytest.param(
Expand Down

0 comments on commit 82e6abe

Please sign in to comment.