LeRobotDataset.__init__() got an unexpected keyword argument 'local_files_only'

[raimmm]

Traceback (most recent call last):
File "/home/ps/Downloads/exercise/OneTwoVLA/scripts/train.py", line 428, in
main(_config.cli())
File "/home/ps/Downloads/exercise/OneTwoVLA/scripts/train.py", line 337, in main
data_loader, val_data_loader = _data_loader.create_data_loader(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/ps/Downloads/exercise/OneTwoVLA/src/openpi/training/data_loader.py", line 167, in create_data_loader
dataset, val_dataset = create_dataset(data_config, config.model)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/ps/Downloads/exercise/OneTwoVLA/src/openpi/training/data_loader.py", line 98, in create_dataset
dataset = umi_dataset.UMIDataset(data_config, model_config.action_horizon)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/ps/Downloads/exercise/OneTwoVLA/src/openpi/policies/umi_dataset.py", line 27, in init
super().init(data_config.repo_id, local_files_only=data_config.local_files_only)
TypeError: LeRobotDataset.init() got an unexpected keyword argument 'local_files_only'

this is my question, I do not know how to solve it.

[hejunqing]

Faced the same problem. I just removed the argument