Skip to content

Commit bd8c91d

Browse files
committed
fixup! style: remove core partition mapper from typing
1 parent 4c47cda commit bd8c91d

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
@@ -8744,7 +8744,7 @@ def test_partitioned_dag_run_with_customized_mapper(
87448744
with dag_maker(
87458745
dag_id="asset-event-consumer",
87468746
schedule=PartitionedAssetTimetable(
8747-
assets=Asset(name="asset-1"),
8747+
assets=asset_1,
87488748
# Most users should use the partition mapper provided by the task-SDK.
87498749
# Advanced users can import from core and register their own partition mapper
87508750
# via an Airflow plugin.
@@ -8822,7 +8822,7 @@ def test_consumer_dag_listen_to_two_partitioned_asset(
88228822
with dag_maker(
88238823
dag_id="asset-event-consumer",
88248824
schedule=PartitionedAssetTimetable(
8825-
assets=(Asset(name="asset-1") & Asset.ref(name="asset-2")),
8825+
assets=(Asset.ref(uri="asset-1") & Asset.ref(name="asset-2")),
88268826
default_partition_mapper=IdentityMapper(),
88278827
),
88288828
session=session,
@@ -8897,7 +8897,7 @@ def test_consumer_dag_listen_to_two_partitioned_asset_with_key_1_mapper(
88978897
with dag_maker(
88988898
dag_id="asset-event-consumer",
88998899
schedule=PartitionedAssetTimetable(
8900-
assets=(Asset(name="asset-1") & Asset(name="asset-2")),
8900+
assets=asset_1 & asset_2,
89018901
# Most users should use the partition mapper provided by the task-SDK.
89028902
# Advanced users can import from core and register their own partition mapper
89038903
# via an Airflow plugin.

0 commit comments

Comments
 (0)