Skip to content

Commit 746682c

Browse files
authored
Chore: Fix the failing clickhouse test (#5339)
1 parent 913414d commit 746682c

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

tests/core/engine_adapter/integration/test_integration.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2630,6 +2630,10 @@ def _mutate_config(current_gateway_name: str, config: Config):
26302630
assert context.default_dialect == "duckdb"
26312631

26322632
schema = ctx.schema(TEST_SCHEMA)
2633+
seed_columns_to_types = {
2634+
"item_id": exp.DataType.build("integer"),
2635+
"event_date": exp.DataType.build("date"),
2636+
}
26332637
seed_query = ctx.input_data(
26342638
pd.DataFrame(
26352639
[
@@ -2643,13 +2647,15 @@ def _mutate_config(current_gateway_name: str, config: Config):
26432647
],
26442648
columns=["item_id", "event_date"],
26452649
),
2646-
columns_to_types={
2647-
"item_id": exp.DataType.build("integer"),
2648-
"event_date": exp.DataType.build("date"),
2649-
},
2650+
columns_to_types=seed_columns_to_types,
26502651
)
26512652
context.upsert_model(
2652-
create_sql_model(name=f"{schema}.seed_model", query=seed_query, kind="FULL")
2653+
create_sql_model(
2654+
name=f"{schema}.seed_model",
2655+
query=seed_query,
2656+
kind="FULL",
2657+
columns=seed_columns_to_types,
2658+
)
26532659
)
26542660

26552661
table_format = ""

0 commit comments

Comments
 (0)