Using WOD 1.3.0 for tutorials

[cool-RR]

I'm just starting out with Waymax, attempting to use the tutorials. I downloaded a few files from the training folder of WOD 1.3.0, but I see that the tutorials data_demo and multi_actors_demo refer to _config.WOD_1_1_0_TRAINING and _config.WOD_1_0_0_VALIDATION respectfully. How do I run with 1.3.0? And which files do I need downloaded to run these 2 tutorials?

[oliverc1623]

Ran into this as well. You have to specify the path in DatasetConfig.

from waymax import config, dataloader


config = config.DatasetConfig(
    path="gs://waymo_open_dataset_motion_v_1_3_0/uncompressed/tf_example/training/training_tfexample.tfrecord-00000-of-01000",
    max_num_objects=32,
)

scenarios = dataloader.simulator_state_generator(config)
scenario = next(scenarios)