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