LICENSE
README.md
pyproject.toml
setup.cfg
src/trajdata/__init__.py
src/trajdata/dataset.py
src/trajdata.egg-info/PKG-INFO
src/trajdata.egg-info/SOURCES.txt
src/trajdata.egg-info/dependency_links.txt
src/trajdata.egg-info/requires.txt
src/trajdata.egg-info/top_level.txt
src/trajdata/augmentation/__init__.py
src/trajdata/augmentation/augmentation.py
src/trajdata/augmentation/low_vel_yaw_correction.py
src/trajdata/augmentation/noise_histories.py
src/trajdata/caching/__init__.py
src/trajdata/caching/df_cache.py
src/trajdata/caching/env_cache.py
src/trajdata/caching/scene_cache.py
src/trajdata/data_structures/__init__.py
src/trajdata/data_structures/agent.py
src/trajdata/data_structures/batch.py
src/trajdata/data_structures/batch_element.py
src/trajdata/data_structures/collation.py
src/trajdata/data_structures/data_index.py
src/trajdata/data_structures/environment.py
src/trajdata/data_structures/scene.py
src/trajdata/data_structures/scene_metadata.py
src/trajdata/data_structures/scene_tag.py
src/trajdata/data_structures/state.py
src/trajdata/dataset_specific/__init__.py
src/trajdata/dataset_specific/raw_dataset.py
src/trajdata/dataset_specific/scene_records.py
src/trajdata/dataset_specific/eth_ucy_peds/__init__.py
src/trajdata/dataset_specific/eth_ucy_peds/eupeds_dataset.py
src/trajdata/dataset_specific/lyft/__init__.py
src/trajdata/dataset_specific/lyft/lyft_dataset.py
src/trajdata/dataset_specific/lyft/lyft_utils.py
src/trajdata/dataset_specific/nuplan/__init__.py
src/trajdata/dataset_specific/nuplan/nuplan_dataset.py
src/trajdata/dataset_specific/nuplan/nuplan_utils.py
src/trajdata/dataset_specific/nusc/__init__.py
src/trajdata/dataset_specific/nusc/nusc_dataset.py
src/trajdata/dataset_specific/nusc/nusc_utils.py
src/trajdata/filtering/__init__.py
src/trajdata/filtering/filters.py
src/trajdata/maps/__init__.py
src/trajdata/maps/lane_route.py
src/trajdata/maps/map_api.py
src/trajdata/maps/map_kdtree.py
src/trajdata/maps/raster_map.py
src/trajdata/maps/traffic_light_status.py
src/trajdata/maps/vec_map.py
src/trajdata/maps/vec_map_elements.py
src/trajdata/parallel/__init__.py
src/trajdata/parallel/data_preprocessor.py
src/trajdata/proto/__init__.py
src/trajdata/proto/vectorized_map_pb2.py
src/trajdata/simulation/__init__.py
src/trajdata/simulation/sim_cache.py
src/trajdata/simulation/sim_df_cache.py
src/trajdata/simulation/sim_metrics.py
src/trajdata/simulation/sim_scene.py
src/trajdata/simulation/sim_stats.py
src/trajdata/simulation/sim_vis.py
src/trajdata/utils/__init__.py
src/trajdata/utils/agent_utils.py
src/trajdata/utils/arr_utils.py
src/trajdata/utils/batch_utils.py
src/trajdata/utils/df_utils.py
src/trajdata/utils/env_utils.py
src/trajdata/utils/map_utils.py
src/trajdata/utils/parallel_utils.py
src/trajdata/utils/raster_utils.py
src/trajdata/utils/scene_utils.py
src/trajdata/utils/state_utils.py
src/trajdata/utils/string_utils.py
src/trajdata/utils/vis_utils.py
src/trajdata/visualization/__init__.py
src/trajdata/visualization/interactive_animation.py
src/trajdata/visualization/interactive_figure.py
src/trajdata/visualization/interactive_vis.py
src/trajdata/visualization/vis.py
tests/test_batch_conversion.py
tests/test_collation.py
tests/test_datasizes.py
tests/test_description_matching.py
tests/test_state.py
tests/test_traffic_data.py
tests/test_vec_map.py