├── CHANGELOG.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── docs ├── conf.py ├── getting_started.rst ├── index.rst ├── notebooks │ ├── data_demo.ipynb │ ├── datatypes_demo.ipynb │ ├── multi_actors_demo.ipynb │ ├── sdc_paths_demo.ipynb │ └── wosac_submission_via_waymax.ipynb └── requirements.txt ├── setup.py └── waymax ├── __init__.py ├── agents ├── __init__.py ├── actor_core.py ├── actor_core_test.py ├── agent_builder.py ├── constant_speed.py ├── constant_speed_test.py ├── expert.py ├── expert_test.py ├── sim_agent.py ├── sim_agent_test.py ├── waypoint_following_agent.py └── waypoint_following_agent_test.py ├── config.py ├── dataloader ├── __init__.py ├── dataloader_utils.py ├── dataloader_utils_test.py ├── testdata │ └── tfrecord_with_routes ├── womd_dataloader.py ├── womd_dataloader_test.py ├── womd_factories.py ├── womd_factories_internal.py ├── womd_factories_test.py └── womd_utils.py ├── datatypes ├── __init__.py ├── action.py ├── array.py ├── array_test.py ├── constant.py ├── object_state.py ├── object_state_test.py ├── observation.py ├── observation_test.py ├── operations.py ├── operations_test.py ├── roadgraph.py ├── roadgraph_test.py ├── route.py ├── route_test.py ├── simulator_state.py ├── simulator_state_test.py ├── traffic_lights.py └── traffic_lights_test.py ├── dynamics ├── README.md ├── __init__.py ├── abstract_dynamics.py ├── abstract_dynamics_test.py ├── bicycle_model.py ├── bicycle_model_test.py ├── delta.py ├── delta_test.py ├── discretizer.py ├── discretizer_test.py ├── state_dynamics.py └── state_dynamics_test.py ├── env ├── __init__.py ├── abstract_environment.py ├── abstract_environment_test.py ├── base_environment.py ├── base_environment_test.py ├── errors.py ├── planning_agent_environment.py ├── planning_agent_environment_test.py ├── rollout.py ├── rollout_test.py ├── typedefs.py └── wrappers │ ├── __init__.py │ ├── brax_wrapper.py │ ├── brax_wrapper_test.py │ ├── dm_env_wrapper.py │ └── dm_env_wrapper_test.py ├── metrics ├── __init__.py ├── abstract_metric.py ├── abstract_metric_test.py ├── comfort.py ├── comfort_test.py ├── imitation.py ├── imitation_test.py ├── metric_factory.py ├── metric_factory_test.py ├── overlap.py ├── overlap_test.py ├── roadgraph.py ├── roadgraph_test.py ├── route.py └── route_test.py ├── rewards ├── __init__.py ├── abstract_reward_function.py ├── abstract_reward_function_test.py ├── linear_combination_reward.py └── linear_combination_reward_test.py ├── utils ├── __init__.py ├── geometry.py ├── geometry_test.py └── test_utils.py └── visualization ├── __init__.py ├── color.py ├── utils.py ├── viz.py └── viz_test.py /CHANGELOG.md: -------------------------------------------------------------------------------- 1 | # Change log 2 | 3 | ## waymax 0.1.0 4 | 5 | Initial release. 6 | -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/README.md -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/getting_started.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/docs/getting_started.rst -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/notebooks/data_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/docs/notebooks/data_demo.ipynb -------------------------------------------------------------------------------- /docs/notebooks/datatypes_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/docs/notebooks/datatypes_demo.ipynb -------------------------------------------------------------------------------- /docs/notebooks/multi_actors_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/docs/notebooks/multi_actors_demo.ipynb -------------------------------------------------------------------------------- /docs/notebooks/sdc_paths_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/docs/notebooks/sdc_paths_demo.ipynb -------------------------------------------------------------------------------- /docs/notebooks/wosac_submission_via_waymax.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/docs/notebooks/wosac_submission_via_waymax.ipynb -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/setup.py -------------------------------------------------------------------------------- /waymax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/__init__.py -------------------------------------------------------------------------------- /waymax/agents/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/agents/__init__.py -------------------------------------------------------------------------------- /waymax/agents/actor_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/agents/actor_core.py -------------------------------------------------------------------------------- /waymax/agents/actor_core_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/agents/actor_core_test.py -------------------------------------------------------------------------------- /waymax/agents/agent_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/agents/agent_builder.py -------------------------------------------------------------------------------- /waymax/agents/constant_speed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/agents/constant_speed.py -------------------------------------------------------------------------------- /waymax/agents/constant_speed_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/agents/constant_speed_test.py -------------------------------------------------------------------------------- /waymax/agents/expert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/agents/expert.py -------------------------------------------------------------------------------- /waymax/agents/expert_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/agents/expert_test.py -------------------------------------------------------------------------------- /waymax/agents/sim_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/agents/sim_agent.py -------------------------------------------------------------------------------- /waymax/agents/sim_agent_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/agents/sim_agent_test.py -------------------------------------------------------------------------------- /waymax/agents/waypoint_following_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/agents/waypoint_following_agent.py -------------------------------------------------------------------------------- /waymax/agents/waypoint_following_agent_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/agents/waypoint_following_agent_test.py -------------------------------------------------------------------------------- /waymax/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/config.py -------------------------------------------------------------------------------- /waymax/dataloader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dataloader/__init__.py -------------------------------------------------------------------------------- /waymax/dataloader/dataloader_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dataloader/dataloader_utils.py -------------------------------------------------------------------------------- /waymax/dataloader/dataloader_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dataloader/dataloader_utils_test.py -------------------------------------------------------------------------------- /waymax/dataloader/testdata/tfrecord_with_routes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dataloader/testdata/tfrecord_with_routes -------------------------------------------------------------------------------- /waymax/dataloader/womd_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dataloader/womd_dataloader.py -------------------------------------------------------------------------------- /waymax/dataloader/womd_dataloader_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dataloader/womd_dataloader_test.py -------------------------------------------------------------------------------- /waymax/dataloader/womd_factories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dataloader/womd_factories.py -------------------------------------------------------------------------------- /waymax/dataloader/womd_factories_internal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dataloader/womd_factories_internal.py -------------------------------------------------------------------------------- /waymax/dataloader/womd_factories_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dataloader/womd_factories_test.py -------------------------------------------------------------------------------- /waymax/dataloader/womd_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dataloader/womd_utils.py -------------------------------------------------------------------------------- /waymax/datatypes/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/__init__.py -------------------------------------------------------------------------------- /waymax/datatypes/action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/action.py -------------------------------------------------------------------------------- /waymax/datatypes/array.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/array.py -------------------------------------------------------------------------------- /waymax/datatypes/array_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/array_test.py -------------------------------------------------------------------------------- /waymax/datatypes/constant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/constant.py -------------------------------------------------------------------------------- /waymax/datatypes/object_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/object_state.py -------------------------------------------------------------------------------- /waymax/datatypes/object_state_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/object_state_test.py -------------------------------------------------------------------------------- /waymax/datatypes/observation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/observation.py -------------------------------------------------------------------------------- /waymax/datatypes/observation_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/observation_test.py -------------------------------------------------------------------------------- /waymax/datatypes/operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/operations.py -------------------------------------------------------------------------------- /waymax/datatypes/operations_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/operations_test.py -------------------------------------------------------------------------------- /waymax/datatypes/roadgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/roadgraph.py -------------------------------------------------------------------------------- /waymax/datatypes/roadgraph_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/roadgraph_test.py -------------------------------------------------------------------------------- /waymax/datatypes/route.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/route.py -------------------------------------------------------------------------------- /waymax/datatypes/route_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/route_test.py -------------------------------------------------------------------------------- /waymax/datatypes/simulator_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/simulator_state.py -------------------------------------------------------------------------------- /waymax/datatypes/simulator_state_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/simulator_state_test.py -------------------------------------------------------------------------------- /waymax/datatypes/traffic_lights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/traffic_lights.py -------------------------------------------------------------------------------- /waymax/datatypes/traffic_lights_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/datatypes/traffic_lights_test.py -------------------------------------------------------------------------------- /waymax/dynamics/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dynamics/README.md -------------------------------------------------------------------------------- /waymax/dynamics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dynamics/__init__.py -------------------------------------------------------------------------------- /waymax/dynamics/abstract_dynamics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dynamics/abstract_dynamics.py -------------------------------------------------------------------------------- /waymax/dynamics/abstract_dynamics_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dynamics/abstract_dynamics_test.py -------------------------------------------------------------------------------- /waymax/dynamics/bicycle_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dynamics/bicycle_model.py -------------------------------------------------------------------------------- /waymax/dynamics/bicycle_model_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dynamics/bicycle_model_test.py -------------------------------------------------------------------------------- /waymax/dynamics/delta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dynamics/delta.py -------------------------------------------------------------------------------- /waymax/dynamics/delta_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dynamics/delta_test.py -------------------------------------------------------------------------------- /waymax/dynamics/discretizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dynamics/discretizer.py -------------------------------------------------------------------------------- /waymax/dynamics/discretizer_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dynamics/discretizer_test.py -------------------------------------------------------------------------------- /waymax/dynamics/state_dynamics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dynamics/state_dynamics.py -------------------------------------------------------------------------------- /waymax/dynamics/state_dynamics_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/dynamics/state_dynamics_test.py -------------------------------------------------------------------------------- /waymax/env/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/__init__.py -------------------------------------------------------------------------------- /waymax/env/abstract_environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/abstract_environment.py -------------------------------------------------------------------------------- /waymax/env/abstract_environment_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/abstract_environment_test.py -------------------------------------------------------------------------------- /waymax/env/base_environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/base_environment.py -------------------------------------------------------------------------------- /waymax/env/base_environment_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/base_environment_test.py -------------------------------------------------------------------------------- /waymax/env/errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/errors.py -------------------------------------------------------------------------------- /waymax/env/planning_agent_environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/planning_agent_environment.py -------------------------------------------------------------------------------- /waymax/env/planning_agent_environment_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/planning_agent_environment_test.py -------------------------------------------------------------------------------- /waymax/env/rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/rollout.py -------------------------------------------------------------------------------- /waymax/env/rollout_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/rollout_test.py -------------------------------------------------------------------------------- /waymax/env/typedefs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/typedefs.py -------------------------------------------------------------------------------- /waymax/env/wrappers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/wrappers/__init__.py -------------------------------------------------------------------------------- /waymax/env/wrappers/brax_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/wrappers/brax_wrapper.py -------------------------------------------------------------------------------- /waymax/env/wrappers/brax_wrapper_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/wrappers/brax_wrapper_test.py -------------------------------------------------------------------------------- /waymax/env/wrappers/dm_env_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/wrappers/dm_env_wrapper.py -------------------------------------------------------------------------------- /waymax/env/wrappers/dm_env_wrapper_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/env/wrappers/dm_env_wrapper_test.py -------------------------------------------------------------------------------- /waymax/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/metrics/__init__.py -------------------------------------------------------------------------------- /waymax/metrics/abstract_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/metrics/abstract_metric.py -------------------------------------------------------------------------------- /waymax/metrics/abstract_metric_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/metrics/abstract_metric_test.py -------------------------------------------------------------------------------- /waymax/metrics/comfort.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/metrics/comfort.py -------------------------------------------------------------------------------- /waymax/metrics/comfort_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/metrics/comfort_test.py -------------------------------------------------------------------------------- /waymax/metrics/imitation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/metrics/imitation.py -------------------------------------------------------------------------------- /waymax/metrics/imitation_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/metrics/imitation_test.py -------------------------------------------------------------------------------- /waymax/metrics/metric_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/metrics/metric_factory.py -------------------------------------------------------------------------------- /waymax/metrics/metric_factory_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/metrics/metric_factory_test.py -------------------------------------------------------------------------------- /waymax/metrics/overlap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/metrics/overlap.py -------------------------------------------------------------------------------- /waymax/metrics/overlap_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/metrics/overlap_test.py -------------------------------------------------------------------------------- /waymax/metrics/roadgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/metrics/roadgraph.py -------------------------------------------------------------------------------- /waymax/metrics/roadgraph_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/metrics/roadgraph_test.py -------------------------------------------------------------------------------- /waymax/metrics/route.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/metrics/route.py -------------------------------------------------------------------------------- /waymax/metrics/route_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/metrics/route_test.py -------------------------------------------------------------------------------- /waymax/rewards/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/rewards/__init__.py -------------------------------------------------------------------------------- /waymax/rewards/abstract_reward_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/rewards/abstract_reward_function.py -------------------------------------------------------------------------------- /waymax/rewards/abstract_reward_function_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/rewards/abstract_reward_function_test.py -------------------------------------------------------------------------------- /waymax/rewards/linear_combination_reward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/rewards/linear_combination_reward.py -------------------------------------------------------------------------------- /waymax/rewards/linear_combination_reward_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/rewards/linear_combination_reward_test.py -------------------------------------------------------------------------------- /waymax/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/utils/__init__.py -------------------------------------------------------------------------------- /waymax/utils/geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/utils/geometry.py -------------------------------------------------------------------------------- /waymax/utils/geometry_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/utils/geometry_test.py -------------------------------------------------------------------------------- /waymax/utils/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/utils/test_utils.py -------------------------------------------------------------------------------- /waymax/visualization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/visualization/__init__.py -------------------------------------------------------------------------------- /waymax/visualization/color.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/visualization/color.py -------------------------------------------------------------------------------- /waymax/visualization/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/visualization/utils.py -------------------------------------------------------------------------------- /waymax/visualization/viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/visualization/viz.py -------------------------------------------------------------------------------- /waymax/visualization/viz_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/waymo-research/waymax/HEAD/waymax/visualization/viz_test.py --------------------------------------------------------------------------------