Skip to content

Commit 8deaae5

Browse files
committed
Fix: Airflow integration test by assigning a change category correctly
1 parent 13000d6 commit 8deaae5

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/schedulers/airflow/test_integration.py

+2
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@ def test_mult_snapshots_same_version(airflow_client: AirflowClient, make_snapsho
7777
model_name = random_name()
7878

7979
snapshot = make_snapshot(_create_model(model_name), version="1")
80+
snapshot.change_category = SnapshotChangeCategory.BREAKING
8081
# Presetting the interval here to avoid backfill.
8182
snapshot.add_interval("2022-01-01", "2022-01-01")
8283
snapshot.set_unpaused_ts(now())
@@ -93,6 +94,7 @@ def test_mult_snapshots_same_version(airflow_client: AirflowClient, make_snapsho
9394
new_fingerprint = original_fingerprint.copy(update={"data_hash": "new_snapshot"})
9495

9596
snapshot.fingerprint = new_fingerprint
97+
snapshot.change_category = SnapshotChangeCategory.FORWARD_ONLY
9698
environment.previous_plan_id = environment.plan_id
9799
environment.plan_id = "new_plan_id"
98100
_apply_plan_and_block(airflow_client, [snapshot], environment)

0 commit comments

Comments
 (0)