Skip to content

Commit

Permalink
FIX skip flaky test of kernel_density in nightly job of CUDA 12.0.1
Browse files Browse the repository at this point in the history
  • Loading branch information
dantegd committed Dec 16, 2024
1 parent 7211507 commit 51cad4d
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion python/cuml/cuml/tests/test_kernel_density.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#
# Copyright (c) 2022-2023, NVIDIA CORPORATION.
# Copyright (c) 2022-2024, NVIDIA CORPORATION.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand All @@ -16,6 +16,7 @@

from cuml.testing.utils import as_type
from sklearn.model_selection import GridSearchCV
import os
import pytest
from hypothesis.extra.numpy import arrays
from hypothesis import given, settings, assume, strategies as st
Expand Down Expand Up @@ -81,6 +82,10 @@ def array_strategy(draw):
)


@pytest.mark.skipif(
os.environ.get("CUDA_VERSION") == "12.0.1",
reason="Flaky hypothesis CI failure: https://github.com/rapidsai/cuml/issues/6173",
)
@settings(deadline=None)
@given(
array_strategy(),
Expand Down

0 comments on commit 51cad4d

Please sign in to comment.