Print warning message if pyarrow is not installed

databricks · Nov 13, 2024 · 9c926bf · 9c926bf
1 parent 608d237
commit 9c926bf
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/databricks/sql/client.py b/src/databricks/sql/client.py
@@ -52,6 +52,12 @@
 
 logger = logging.getLogger(__name__)
 
+if pyarrow is None:
+    logger.warning(
+        "[WARN] pyarrow is not installed, any arrow specific api (e.g. fetchmany_arrow) and cloud fetch will be disabled."
+        "If you need these features, please run pip install pyarrow to install"
+    )
+
 DEFAULT_RESULT_BUFFER_SIZE_BYTES = 104857600
 DEFAULT_ARRAY_SIZE = 100000