Skip to content

Commit 8ec2ee0

Browse files
committed
fixup! style: remove core partition mapper from typing
1 parent 60a4b6a commit 8ec2ee0

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
@@ -8747,7 +8747,7 @@ def test_partitioned_dag_run_with_customized_mapper(
87478747
with dag_maker(
87488748
dag_id="asset-event-consumer",
87498749
schedule=PartitionedAssetTimetable(
8750-
assets=Asset(name="asset-1"),
8750+
assets=asset_1,
87518751
# Most users should use the partition mapper provided by the task-SDK.
87528752
# Advanced users can import from core and register their own partition mapper
87538753
# via an Airflow plugin.
@@ -8825,7 +8825,7 @@ def test_consumer_dag_listen_to_two_partitioned_asset(
88258825
with dag_maker(
88268826
dag_id="asset-event-consumer",
88278827
schedule=PartitionedAssetTimetable(
8828-
assets=(Asset(name="asset-1") & Asset.ref(name="asset-2")),
8828+
assets=(Asset.ref(uri="asset-1") & Asset.ref(name="asset-2")),
88298829
default_partition_mapper=IdentityMapper(),
88308830
),
88318831
session=session,
@@ -8900,7 +8900,7 @@ def test_consumer_dag_listen_to_two_partitioned_asset_with_key_1_mapper(
89008900
with dag_maker(
89018901
dag_id="asset-event-consumer",
89028902
schedule=PartitionedAssetTimetable(
8903-
assets=(Asset(name="asset-1") & Asset(name="asset-2")),
8903+
assets=asset_1 & asset_2,
89048904
# Most users should use the partition mapper provided by the task-SDK.
89058905
# Advanced users can import from core and register their own partition mapper
89068906
# via an Airflow plugin.

0 commit comments

Comments
 (0)