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