Skip to content

Commit

Permalink
fixes clickhouse adapter test
Browse files Browse the repository at this point in the history
  • Loading branch information
rudolfix committed Nov 14, 2024
1 parent 3dcf2ef commit 73b79ee
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions tests/load/clickhouse/test_clickhouse_adapter.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
from typing import Generator, Dict, cast

import dlt
from dlt.common.utils import custom_environ
from dlt.destinations.adapters import clickhouse_adapter
from dlt.destinations.impl.clickhouse.sql_client import ClickHouseSqlClient
from dlt.destinations.impl.clickhouse.typing import TDeployment
Expand Down Expand Up @@ -34,7 +33,12 @@ def not_annotated_resource() -> Generator[Dict[str, int], None, None]:
clickhouse_adapter(merge_tree_resource, table_engine_type="merge_tree")
clickhouse_adapter(replicated_merge_tree_resource, table_engine_type="replicated_merge_tree")

pipe = dlt.pipeline(pipeline_name="adapter_test", destination="clickhouse", dev_mode=True)
pipe = dlt.pipeline(
pipeline_name="adapter_test",
destination="clickhouse",
dev_mode=True,
dataset_name="adapter_test_ds",
)

with pipe.sql_client() as client:
deployment_type: TDeployment = get_deployment_type(cast(ClickHouseSqlClient, client))
Expand Down

0 comments on commit 73b79ee

Please sign in to comment.