diff --git a/tests/unittests/classification/test_stat_scores.py b/tests/unittests/classification/test_stat_scores.py index 53fa78d0368..b9b10927d4c 100644 --- a/tests/unittests/classification/test_stat_scores.py +++ b/tests/unittests/classification/test_stat_scores.py @@ -11,7 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. - +import sys from functools import partial import numpy as np @@ -578,6 +578,7 @@ def test_multilabel_stat_scores_dtype_gpu(self, inputs, dtype): ) +@pytest.mark.flaky(retries=3, condition=sys.platform.startswith("win32")) def test_support_for_int(): """See issue: https://github.com/Lightning-AI/torchmetrics/issues/1970.""" seed_all(42)