From b4fb6d1177f65ca3ff4c3137841d6603d45b5e00 Mon Sep 17 00:00:00 2001 From: Guillaume Wenzek Date: Wed, 4 Oct 2023 10:56:33 +0200 Subject: [PATCH] hide test case --- tests/unit/data/test_collater.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/unit/data/test_collater.py b/tests/unit/data/test_collater.py index 9082988ef..c2cc13c08 100644 --- a/tests/unit/data/test_collater.py +++ b/tests/unit/data/test_collater.py @@ -9,7 +9,7 @@ from torch.nn.functional import pad from fairseq2.data import CollateOptionsOverride, Collater, read_sequence -from tests.common import assert_close, assert_equal, device +from tests.common import assert_close, assert_equal, device, python_devel_only class TestCollater: @@ -380,6 +380,7 @@ def test_init_raises_error_when_pad_idx_is_none_and_pad_to_multiple_is_greater_t Collater(overrides=[CollateOptionsOverride("foo", pad_to_multiple=2)]) +@pytest.mark.skipif(python_devel_only(), reason="fairseq2n 0.2.0") @pytest.mark.parametrize("pad_to_multiple,pad_size", [(1, 0), (2, 0), (3, 2), (8, 4)]) def test_collate_works_when_input_has_sequence_tensors( pad_to_multiple: int, pad_size: int