Skip to content

Commit 8c9d2ed

Browse files
committed
fixup! style: remove core partition mapper from typing
1 parent 3ad20d7 commit 8c9d2ed

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

airflow-core/tests/unit/jobs/test_scheduler_job.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8798,7 +8798,7 @@ def test_partitioned_dag_run_with_customized_mapper(
87988798
with dag_maker(
87998799
dag_id="asset-event-consumer",
88008800
schedule=PartitionedAssetTimetable(
8801-
assets=Asset(name="asset-1"),
8801+
assets=asset_1,
88028802
# Most users should use the partition mapper provided by the task-SDK.
88038803
# Advanced users can import from core and register their own partition mapper
88048804
# via an Airflow plugin.
@@ -8852,7 +8852,7 @@ def test_consumer_dag_listen_to_two_partitioned_asset(
88528852
with dag_maker(
88538853
dag_id="asset-event-consumer",
88548854
schedule=PartitionedAssetTimetable(
8855-
assets=(Asset(name="asset-1") & Asset.ref(name="asset-2")),
8855+
assets=(Asset.ref(uri="asset-1") & Asset.ref(name="asset-2")),
88568856
default_partition_mapper=IdentityMapper(),
88578857
),
88588858
session=session,
@@ -8934,7 +8934,7 @@ def test_consumer_dag_listen_to_two_partitioned_asset_with_key_1_mapper(
89348934
with dag_maker(
89358935
dag_id="asset-event-consumer",
89368936
schedule=PartitionedAssetTimetable(
8937-
assets=(Asset(name="asset-1") & Asset(name="asset-2")),
8937+
assets=asset_1 & asset_2,
89388938
# Most users should use the partition mapper provided by the task-SDK.
89398939
# Advanced users can import from core and register their own partition mapper
89408940
# via an Airflow plugin.

0 commit comments

Comments
 (0)