fixes for 0.5.1

dlt-hub · Jul 8, 2024 · 692aa0c · 692aa0c
1 parent cbab9ed
commit 692aa0c
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/tests/pg_replication/conftest.py b/tests/pg_replication/conftest.py
@@ -17,7 +17,7 @@ def src_config() -> Iterator[Tuple[dlt.Pipeline, str, str]]:
         destination=dlt.destinations.postgres(
             credentials=dlt.secrets.get("sources.pg_replication.credentials")
         ),
-        full_refresh=True,
+        dev_mode=True,
     )
     yield src_pl, slot, pub
     # teardown
@@ -27,7 +27,7 @@ def src_config() -> Iterator[Tuple[dlt.Pipeline, str, str]]:
             c.drop_dataset()
         except Exception as e:
             print(e)
-        with c.with_staging_dataset(staging=True):
+        with c.with_staging_dataset():
             try:
                 c.drop_dataset()
             except Exception as e:

diff --git a/tests/sql_database/test_sql_database_source.py b/tests/sql_database/test_sql_database_source.py
@@ -60,11 +60,11 @@ def convert_time_to_us(table):
     import pyarrow as pa
     from pyarrow import compute as pc
 
-    time_ns_column = table['time_col']
-    time_us_column = pc.cast(time_ns_column, pa.time64('us'), safe=False)
+    time_ns_column = table["time_col"]
+    time_us_column = pc.cast(time_ns_column, pa.time64("us"), safe=False)
     new_table = table.set_column(
-        table.column_names.index('time_col'),
-        'time_col',
+        table.column_names.index("time_col"),
+        "time_col",
         time_us_column,
     )
     return new_table

diff --git a/tests/utils.py b/tests/utils.py
@@ -108,7 +108,7 @@ def _drop_dataset(schema_name: str) -> None:
                         # print("dropped")
                     except Exception as exc:
                         print(exc)
-                    with c.with_staging_dataset(staging=True):
+                    with c.with_staging_dataset():
                         try:
                             c.drop_dataset()
                             # print("dropped")