From de3a7b5780ea4855c76a44f3a5b5d79fe0810635 Mon Sep 17 00:00:00 2001 From: Vedant-Mhatre Date: Sat, 21 Feb 2026 19:58:16 -0800 Subject: [PATCH] Use WOD 1.3.0 configs in tutorial notebooks --- docs/notebooks/data_demo.ipynb | 6 +++--- docs/notebooks/multi_actors_demo.ipynb | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/docs/notebooks/data_demo.ipynb b/docs/notebooks/data_demo.ipynb index 1bcfb35..d89a52d 100644 --- a/docs/notebooks/data_demo.ipynb +++ b/docs/notebooks/data_demo.ipynb @@ -38,9 +38,9 @@ }, "source": [ "\n", - "We first create a dataset config, using the default configs provided in the `waymax.config` module. In particular, `config.WOD_1_1_0_TRAINING` is a pre-defined configuration that points to version 1.1.0 of the Waymo Open Dataset.\n", + "We first create a dataset config, using the default configs provided in the `waymax.config` module. In particular, `config.WOD_1_3_0_TRAINING` is a pre-defined configuration that points to version 1.3.0 of the Waymo Open Dataset.\n", "\n", - "The data config contains a number of options to configure how and where the dataset is loaded from. By default, the `WOD_1_1_0_TRAINING` loads up to 128 objects (e.g. vehicles, pedestrians) per scenario. Here, we can save memory and compute by loading only the first 32 objects stored in the scenario.\n", + "The data config contains a number of options to configure how and where the dataset is loaded from. By default, the `WOD_1_3_0_TRAINING` loads up to 128 objects (e.g. vehicles, pedestrians) per scenario. Here, we can save memory and compute by loading only the first 32 objects stored in the scenario.\n", "\n", "We use the `dataloader.simulator_state_generator` function to create an iterator\n", "through Open Motion Dataset scenarios. Calling next on the iterator will retrieve the first scenario in the dataset.\n" @@ -54,7 +54,7 @@ }, "outputs": [], "source": [ - "config = dataclasses.replace(_config.WOD_1_1_0_TRAINING, max_num_objects=32)\n", + "config = dataclasses.replace(_config.WOD_1_3_0_TRAINING, max_num_objects=32)\n", "data_iter = dataloader.simulator_state_generator(config=config)\n", "scenario = next(data_iter)" ] diff --git a/docs/notebooks/multi_actors_demo.ipynb b/docs/notebooks/multi_actors_demo.ipynb index 4a65f2d..674356e 100644 --- a/docs/notebooks/multi_actors_demo.ipynb +++ b/docs/notebooks/multi_actors_demo.ipynb @@ -47,7 +47,7 @@ "max_num_objects = 32\n", "\n", "config = dataclasses.replace(\n", - " _config.WOD_1_0_0_VALIDATION, max_num_objects=max_num_objects\n", + " _config.WOD_1_3_0_VALIDATION, max_num_objects=max_num_objects\n", ")\n", "data_iter = dataloader.simulator_state_generator(config=config)\n", "scenario = next(data_iter)"