@@ -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