This commit is contained in:
Cadene 2024-04-15 21:28:34 +00:00
parent 67d79732f9
commit c7a8218620
1 changed files with 1 additions and 1 deletions

View File

@ -152,7 +152,7 @@ def add_episodes_inplace(data_dict, online_dataset, concat_dataset, sampler, pc_
example["episode_data_id_to"] += start_index example["episode_data_id_to"] += start_index
return example return example
disable_progress_bar() # map has a tqdm progres bar disable_progress_bar() # map has a tqdm progress bar
data_dict = data_dict.map(shift_indices) data_dict = data_dict.map(shift_indices)
# extend online dataset # extend online dataset