Ignroe deprecation warnings related to n_clusters

rapidsai · Jan 16, 2025 · d2b96e2 · d2b96e2
1 parent ed127a9
commit d2b96e2
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 8 deletions.
diff --git a/python/cuml/cuml/tests/experimental/accel/estimators_hyperparams/test_accel_umap.py b/python/cuml/cuml/tests/experimental/accel/estimators_hyperparams/test_accel_umap.py
@@ -21,10 +21,16 @@
 from sklearn.manifold import trustworthiness
 
 
-pytestmark = pytest.mark.filterwarnings(
-    "ignore:The default value of `data_on_host` "
-    'will change from False to "auto" in 25.06'
-)
+pytestmark = [
+    pytest.mark.filterwarnings(
+        "ignore:The default value of `data_on_host` "
+        'will change from False to "auto" in 25.06'
+    ),
+    pytest.mark.filterwarnings(
+        "ignore:The default value of `nnd_n_clusters` "
+        "will change from 1 to 10 in 25.06."
+    ),
+]
 
 
 @pytest.fixture(scope="module")

diff --git a/python/cuml/cuml/tests/test_umap.py b/python/cuml/cuml/tests/test_umap.py
@@ -50,10 +50,16 @@
 if not IS_ARM:
     import umap
 
-pytestmark = pytest.mark.filterwarnings(
-    "ignore:The default value of `data_on_host` "
-    'will change from False to "auto" in 25.06'
-)
+pytestmark = [
+    pytest.mark.filterwarnings(
+        "ignore:The default value of `data_on_host` "
+        'will change from False to "auto" in 25.06'
+    ),
+    pytest.mark.filterwarnings(
+        "ignore:The default value of `nnd_n_clusters` "
+        "will change from 1 to 10 in 25.06."
+    ),
+]
 
 dataset_names = ["iris", "digits", "wine", "blobs"]