"Partitioned Runs" tab is missing on UI

[atul-astronomer]

Apache Airflow version

main (development)

If "Other Airflow 3 version" selected, which one?

No response

What happened?

Asset partitioned dagruns tab should be displayed on UI for consumer dags.

What you think should happen instead?

No response

How to reproduce

Use the below Dag:

from __future__ import annotations

from airflow import DAG
from airflow.providers.standard.operators.empty import EmptyOperator
from airflow.sdk import (
    Asset,
    PartitionedAssetTimetable,
    IdentityMapper,
)

from datetime import datetime

sales_data = Asset(uri="file://aip76/sales_data.csv")
inventory_data = Asset(uri="file://aip76/inventory_data.csv")
pricing_data = Asset(uri="file://aip76/pricing_data.csv")


with DAG(
    dag_id="ui_partition_asset_producer",
    start_date=datetime(2024, 1, 1),
    schedule=None,   # manual partition control
    catchup=False,
    tags=["aip76", "ui_test"],
) as producer_dag:

    produce_sales = EmptyOperator(
        task_id="produce_sales",
        outlets=[sales_data],
    )

    produce_inventory = EmptyOperator(
        task_id="produce_inventory",
        outlets=[inventory_data],
    )

    produce_pricing = EmptyOperator(
        task_id="produce_pricing",
        outlets=[pricing_data],
    )


with DAG(
    dag_id="ui_partition_asset_consumer",
    start_date=datetime(2024, 1, 1),
    schedule=PartitionedAssetTimetable(
        assets=sales_data & inventory_data & pricing_data,
        default_partition_mapper=IdentityMapper(),
    ),
    catchup=False,
    tags=["aip76", "ui_test"],
) as consumer_dag:

    consume_task = EmptyOperator(
        task_id="consume_all_assets"
    )

Operating System

Linux

Versions of Apache Airflow Providers

No response

Deployment

Other

Deployment details

No response

Anything else?

No response

Are you willing to submit PR?

• Yes I am willing to submit a PR!

Code of Conduct

• I agree to follow this project's Code of Conduct