Skip to content
5 changes: 5 additions & 0 deletions dask_sql/context.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,11 @@
from dask.base import optimize
from dask.distributed import Client

try:
import dask_cuda # noqa: F401
except ImportError: # pragma: no cover
pass

from dask_sql import input_utils
from dask_sql.datacontainer import (
UDF,
Expand Down
4 changes: 4 additions & 0 deletions tests/integration/fixtures.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,10 @@

try:
import cudf

# importing to check for JVM segfault
import dask_cudf # noqa: F401
from dask_cuda import LocalCUDACluster # noqa: F401
except ImportError:
cudf = None

Expand Down