Skip to content
7 changes: 7 additions & 0 deletions dask_sql/context.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,13 @@
from dask_sql.physical.rex import RexConverter, core
from dask_sql.utils import ParsingException

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


if TYPE_CHECKING:
from dask_sql.java import org

Expand Down
10 changes: 10 additions & 0 deletions tests/integration/fixtures.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,16 @@
except ImportError:
cudf = None

try:
import dask_cudf
except ImportError:
dask_cudf = None

try:
from dask_cuda import LocalCUDACluster # noqa: F401
except ImportError:
dask_cuda = None


@pytest.fixture()
def timeseries_df(c):
Expand Down