From 9c926bf599abbd563aaaa1df07f8edb35d20d726 Mon Sep 17 00:00:00 2001
From: Jacky Hu <jacky.hu@databricks.com>
Date: Tue, 12 Nov 2024 18:54:23 -0800
Subject: [PATCH] Print warning message if pyarrow is not installed

---
 src/databricks/sql/client.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/databricks/sql/client.py b/src/databricks/sql/client.py
index 4e0ab941..08e47c5c 100755
--- 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