Fix tests

dlt-hub · Aug 26, 2024 · 39d7859 · 39d7859
1 parent 227c89c
commit 39d7859
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/dlt/destinations/impl/bigquery/bigquery.py b/dlt/destinations/impl/bigquery/bigquery.py
@@ -229,9 +229,7 @@ def __init__(
             config.http_timeout,
             config.retry_deadline,
         )
-        SupportsStagingDestination.__init__(
-            self, config.truncate_table_before_load_on_staging_destination
-        )
+        SupportsStagingDestination.__init__(self, config)
         super().__init__(schema, config, sql_client)
         self.config: BigQueryClientConfiguration = config
         self.sql_client: BigQuerySqlClient = sql_client  # type: ignore

diff --git a/dlt/destinations/impl/dummy/dummy.py b/dlt/destinations/impl/dummy/dummy.py
@@ -134,7 +134,7 @@ def __init__(
         config: DummyClientConfiguration,
         capabilities: DestinationCapabilitiesContext,
     ) -> None:
-        SupportsStagingDestination.__init__(self, config)
+        SupportsStagingDestination.__init__(self, config)  # type: ignore
         super().__init__(schema, config, capabilities)
         self.in_staging_context = False
         self.config: DummyClientConfiguration = config

diff --git a/tests/load/test_dummy_client.py b/tests/load/test_dummy_client.py
@@ -558,7 +558,7 @@ def test_truncate_table_before_load_on_stanging(to_truncate) -> None:
     load_id, schema = prepare_load_package(load.load_storage, NORMALIZED_FILES)
     destination_client = load.get_destination_client(schema)
     assert (
-        destination_client.should_truncate_table_before_load_on_staging_destination(
+        destination_client.should_truncate_table_before_load_on_staging_destination(  # type: ignore
             schema.tables["_dlt_version"]
         )
         == to_truncate