import mmcv
from typing import Dict, List

# predroot = '/mnt/hdd/models/UniAD/stage2_e2e/017_carla_closedloop/e2e_carla_slurm_r50_v2t1_paradrive/test/Sun_Apr_14_19_11_00_2024/results.pkl'
predroot = '/mnt/hdd2/datasets/navsim/openscene-v1.1/infos/nuplan_navsim_train.pkl'

outputs: Dict = mmcv.load(predroot)
# keys are bbox_results, occ_results_computed, planning_results_computed

# print(outputs.keys())
# print(outputs[0].keys())
# zxc

# outputs: List[Dict] = outputs["infos"]
# outputs: List[Dict] = outputs["bbox_results"]
# self.outputs = outputs

for index in range(len(outputs)):
    data = outputs[index]
    print('\n\nnew frame')
    print('scene_token ', data['scene_token'])
    print(data['token'])
    print(data['frame_idx'])