fix: allow importing from_table_to_db

pacman82 · Apr 3, 2023 · 57e15c0 · 57e15c0
1 parent 743dddf
commit 57e15c0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/python/arrow_odbc/__init__.py b/python/arrow_odbc/__init__.py
@@ -1,6 +1,6 @@
 from .error import Error
 from .reader import BatchReader, read_arrow_batches_from_odbc
-from .writer import BatchWriter, insert_into_table
+from .writer import BatchWriter, insert_into_table, from_table_to_db
 
 __all__ = [
     "BatchReader",

diff --git a/tests/test_arrow_odbc.py b/tests/test_arrow_odbc.py
@@ -11,8 +11,12 @@
 
 from pytest import raises
 
-from arrow_odbc import read_arrow_batches_from_odbc, Error
-from arrow_odbc.writer import insert_into_table, from_table_to_db
+from arrow_odbc import (
+    insert_into_table,
+    from_table_to_db,
+    read_arrow_batches_from_odbc,
+    Error,
+)
 
 MSSQL = "Driver={ODBC Driver 17 for SQL Server};Server=localhost;UID=SA;PWD=My@Test@Password1;"