Merge branch 'branch-24.04' into reuse-indexing

rapidsai · Mar 3, 2024 · 639f5c4 · 639f5c4
2 parents 5095760 + b0ba340
commit 639f5c4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/cuml/tests/test_strategies.py b/python/cuml/tests/test_strategies.py
@@ -1,4 +1,4 @@
-# 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.
@@ -98,7 +98,9 @@ def test_standard_datasets(dataset):
 
 
 @given(split_datasets(standard_datasets()))
-@settings(suppress_health_check=[HealthCheck.too_slow])
+@settings(
+    suppress_health_check=[HealthCheck.too_slow, HealthCheck.data_too_large]
+)
 def test_split_datasets(split_dataset):
     X_train, X_test, y_train, y_test = split_dataset