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)"