Error about creating Waymo dataset info file

[mengmeng4263]

Hi PointAugmenting authors,

Thanks for sharing your code. I have encountered the error when trying to create Waymo dataset info file by using this command python tools/create_data.py waymo_data_prep --root_path=data/Waymo --split train --nsweeps=1

Traceback (most recent call last):
File "tools/create_data.py", line 36, in
fire.Fire()
File "/opt/conda/lib/python3.7/site-packages/fire/core.py", line 141, in Fire
component_trace = _Fire(component, args, parsed_flag_args, context, name)
File "/opt/conda/lib/python3.7/site-packages/fire/core.py", line 471, in _Fire
target=component.name)
File "/opt/conda/lib/python3.7/site-packages/fire/core.py", line 681, in _CallAndUpdateTrace
component = fn(*varargs, **kwargs)
File "tools/create_data.py", line 29, in waymo_data_prep
nsweeps=nsweeps
File "/barn3/mengmeng/PointAugmenting/det3d/datasets/utils/create_gt_database.py", line 91, in create_groundtruth_database
gt_frustums = annos["frustums"]
KeyError: 'frustums'