WOSAC eval locally becomes very slow

[pengzhenghao]

Hi,

I notice that the WOSAC eval code becomes very slow after the update to new version of waymo-dataset.

My version:

waymo-open-dataset-tf-2-12-0 1.6.4

The profile information:

My code entering waymo_dataset's functions:

        config = load_metrics_config(use_2024=use_2024)
        scenario_metrics = metrics.compute_scenario_metrics_for_bundle(config, scenario, scenario_rollouts)
        aggregate_metrics = metrics.aggregate_metrics_to_buckets(config, scenario_metrics)

[pengzhenghao]

I should mention that it takes 32s on my PC to eval a single scenario (with 32 modes).