move ibis stuff to helpers

dlt-hub · Dec 6, 2024 · 7af8870 · 7af8870
1 parent f22b1dc
commit 7af8870
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 7 deletions.
diff --git a/dlt/common/destination/reference.py b/dlt/common/destination/reference.py
@@ -76,7 +76,7 @@
     try:
         from dlt.common.libs.pandas import DataFrame
         from dlt.common.libs.pyarrow import Table as ArrowTable
-        from dlt.common.libs.ibis import BaseBackend as IbisBackend
+        from dlt.helpers.ibis import BaseBackend as IbisBackend
     except MissingDependencyException:
         DataFrame = Any
         ArrowTable = Any

diff --git a/dlt/destinations/dataset/dataset.py b/dlt/destinations/dataset/dataset.py
@@ -21,7 +21,7 @@
 
 if TYPE_CHECKING:
     try:
-        from dlt.common.libs.ibis import BaseBackend as IbisBackend
+        from dlt.helpers.ibis import BaseBackend as IbisBackend
     except MissingDependencyException:
         IbisBackend = Any
 else:
@@ -47,7 +47,7 @@ def __init__(
 
     def ibis(self) -> IbisBackend:
         """return a connected ibis backend"""
-        from dlt.common.libs.ibis import create_ibis_backend
+        from dlt.helpers.ibis import create_ibis_backend
 
         self._ensure_client_and_schema()
         return create_ibis_backend(
@@ -117,7 +117,7 @@ def table(self, table_name: str) -> SupportsReadableRelation:
         # we can create an ibis powered relation if ibis is available
         if table_name in self.schema.tables and self._dataset_type in ("auto", "ibis"):
             try:
-                from dlt.common.libs.ibis import create_unbound_ibis_table
+                from dlt.helpers.ibis import create_unbound_ibis_table
                 from dlt.destinations.dataset.ibis_relation import ReadableIbisRelation
 
                 unbound_table = create_unbound_ibis_table(self.sql_client, self.schema, table_name)

diff --git a/dlt/destinations/dataset/ibis_relation.py b/dlt/destinations/dataset/ibis_relation.py
@@ -13,7 +13,7 @@
     ReadableDBAPIDataset = Any
 
 try:
-    from dlt.common.libs.ibis import Expr
+    from dlt.helpers.ibis import Expr
 except MissingDependencyException:
     Expr = Any
 
@@ -63,7 +63,7 @@ def __init__(
     def query(self) -> Any:
         """build the query"""
 
-        from dlt.common.libs.ibis import ibis, sqlglot
+        from dlt.helpers.ibis import ibis, sqlglot
 
         destination_type = self._dataset._destination.destination_type
         target_dialect = DIALECT_MAP[destination_type]

diff --git a/dlt/common/libs/ibis.py → dlt/helpers/ibis.py b/dlt/common/libs/ibis.py → dlt/helpers/ibis.py
diff --git a/tests/load/test_read_interfaces.py b/tests/load/test_read_interfaces.py
@@ -633,7 +633,7 @@ def sql_from_expr(expr: Any) -> str:
 def test_ibis_dataset_access(populated_pipeline: Pipeline) -> None:
     # NOTE: we could generalize this with a context for certain deps
 
-    from dlt.common.libs.ibis import SUPPORTED_DESTINATIONS
+    from dlt.helpers.ibis import SUPPORTED_DESTINATIONS
 
     # check correct error if not supported
     if populated_pipeline.destination.destination_type not in SUPPORTED_DESTINATIONS: