Bug in calculating advantage script

[akshay18iitg]

System Info

Main branch code

Information

• One of the scripts in the src/opentau/scripts/ folder of OpenTau
• My own task or dataset (give details below)

Reproduction

python src/opentau/scripts/get_advantage_and_percentiles.py --config_path=outputs/train/2026-02-28/04-45-50_value_mlp/checkpoints/024000/train_config.json --batch_size=20 --dataloader_batch_size=20 --dataset_mixture=configs/examples/advantage_config.json

Expected behavior

When i run the script I see the below error
NFO 2026-03-02 04:27:18 dataset.py:1097 Checking timestamps synchronization...
Traceback (most recent call last):
File "/home/autox/akshay/OpenTau/src/opentau/scripts/get_advantage_and_percentiles.py", line 220, in
main()
File "/home/autox/akshay/OpenTau/src/opentau/scripts/get_advantage_and_percentiles.py", line 99, in filtered_wrapper
return wrapped_fn(*args, **kwargs)
File "/home/autox/akshay/OpenTau/src/opentau/configs/parser.py", line 388, in wrapper_inner
response = fn(cfg, *args, **kwargs)
File "/home/autox/akshay/OpenTau/src/opentau/scripts/get_advantage_and_percentiles.py", line 138, in main
assert isinstance(dataset, LeRobotDataset), (
AssertionError: Expected instance of LeRobotDataset, got <class 'tuple'>

[jiangziben]

this is caused by valid dataset