filter pytests only

rapidsai · Feb 12, 2024 · 7c65612 · 7c65612
1 parent 39c0a11
commit 7c65612
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 9 deletions.
diff --git a/dask_cuda/__init__.py b/dask_cuda/__init__.py
@@ -1,17 +1,8 @@
 import sys
-import warnings
 
 if sys.platform != "linux":
     raise ImportError("Only Linux is supported by Dask-CUDA at this time")
 
-# This warning must be filtered until dask-expr support
-# is enabled in both dask-cudf and dask-cuda.
-# See: https://github.com/rapidsai/dask-cuda/issues/1311
-warnings.filterwarnings(
-    "ignore",
-    message="The current Dask DataFrame implementation is deprecated.",
-)
-
 import dask
 import dask.utils
 import dask.dataframe.core

diff --git a/pyproject.toml b/pyproject.toml
@@ -123,6 +123,10 @@ filterwarnings = [
     "error::FutureWarning",
     # remove after https://github.com/rapidsai/dask-cuda/issues/1087 is closed
     "ignore:There is no current event loop:DeprecationWarning:tornado",
+    # This warning must be filtered until dask-expr support
+    # is enabled in both dask-cudf and dask-cuda.
+    # See: https://github.com/rapidsai/dask-cuda/issues/1311
+    "ignore:Dask DataFrame implementation is deprecated:DeprecationWarning",
 ]
 
 [tool.setuptools]