Fix scene mode when using the `standardize_data=False`

NVlabs / trajdata

A unified interface to many trajectory forecasting datasets.

Apache License 2.0

290 stars 38 forks source link

Closed Leon0402 closed 2 years ago

Leon0402 commented 2 years ago

One variable was missing and one wrongly named when using the standardize_data=False, which are required here: https://github.com/nvr-avg/trajdata/blob/main/src/trajdata/data_structures/collation.py#L708

Also fixed some missing / wrong types I found while browsing the code.

BorisIvanovic commented 2 years ago

Whoops, great catches! Thank you for submitting this @Leon0402, all looks good. LGTM!