how to solve this?

[kailushan]

[Job 5 dataset creation] finished in 6700 ms
Job 5: creating dataset... [OK]
job 0 scenes:: 0%| | 0/707 [00:01<?, ?it/s]
multiprocessing.pool.RemoteTraceback:
"""
Traceback (most recent call last):
File "/xxxxx/lib/python3.7/multiprocessing/pool.py", line 121, in worker
result = (True, func(*args, **kwds))
File "prerender_raster.py", line 177, in pre_render_scenes
data = ds.agent_dataset.get_frame(scene_num, state_index=state_index, track_id=track_id)
File "/xxxxx/lib/python3.7/site-packages/l5kit/dataset/ego.py", line 100, in get_frame
image = data["image"].transpose(2, 0, 1)
AttributeError: 'dict' object has no attribute 'transpose'
"""