fix: put _DataframeInfos and _ArrowInfos under TYPE_CHECKING block

sfu-db · Apr 19, 2024 · 7158e80 · 7158e80
1 parent a336e00
commit 7158e80
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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