From 7158e805ab7a6c0e045103baf638b4749e8d0419 Mon Sep 17 00:00:00 2001
From: "ZhengYu, Xu" <zen-xu@outlook.com>
Date: Fri, 19 Apr 2024 15:14:03 +0800
Subject: [PATCH] fix: put `_DataframeInfos` and `_ArrowInfos` under
 TYPE_CHECKING block

---
 connectorx-python/connectorx/__init__.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/connectorx-python/connectorx/__init__.py b/connectorx-python/connectorx/__init__.py
index 456042077..441c67394 100644
--- a/connectorx-python/connectorx/__init__.py
+++ b/connectorx-python/connectorx/__init__.py
@@ -10,8 +10,6 @@
     partition_sql as _partition_sql,
     read_sql2 as _read_sql2,
     get_meta as _get_meta,
-    _DataframeInfos,
-    _ArrowInfos,
 )
 
 if TYPE_CHECKING:
@@ -21,6 +19,10 @@
     import dask.dataframe as dd
     import pyarrow as pa
 
+    # only for typing hints
+    from .connectorx import  _DataframeInfos, _ArrowInfos
+
+
 __version__ = version(__name__)
 
 import os