diff --git a/tpcx_bb/xbb_tools/cluster_startup.py b/tpcx_bb/xbb_tools/cluster_startup.py index 637e6fb3..0be7e302 100644 --- a/tpcx_bb/xbb_tools/cluster_startup.py +++ b/tpcx_bb/xbb_tools/cluster_startup.py @@ -22,7 +22,6 @@ import dask from dask.distributed import Client from dask.utils import parse_bytes -from blazingsql import BlazingContext def get_config_options(): @@ -110,6 +109,7 @@ def maybe_create_worker_directories(dask_worker): bc = None if create_blazing_context: + from blazingsql import BlazingContext bc = BlazingContext( dask_client=client, pool=os.environ.get("BLAZING_POOL", False),