From 58bd34d8f1620e1b677d39e6bc6dbd18bf34fa0e Mon Sep 17 00:00:00 2001
From: esoha-nvidia <69258779+esoha-nvidia@users.noreply.github.com>
Date: Fri, 6 Nov 2020 12:27:36 -0700
Subject: [PATCH] Only import BlazingSQL as necessary (#109)

BlazingSQL is broken: https://github.com/BlazingDB/blazingsql/issues/1037

This workaround allows the repo to work when BlazingSQL isn't enabled.
---
 tpcx_bb/xbb_tools/cluster_startup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tpcx_bb/xbb_tools/cluster_startup.py b/tpcx_bb/xbb_tools/cluster_startup.py
index e45f2633..b325d2a2 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_bsql_config_options():
@@ -127,6 +126,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),