admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-30 21:19:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(fromat): formatting code

Browse Source
This commit is contained in:
CarolinePascal
2026-07-30 17:32:15 +02:00
parent dca9e80f03
commit e48b395c60
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/dataset_tools.py
+1 -1
View File
@@ -1444,7 +1444,7 @@ def modify_tasks(
2. Set specific tasks for specific episodes (using `episode_tasks`) 2. Set specific tasks for specific episodes (using `episode_tasks`)
3. Replace existing task strings wherever they appear (using `task_replacements`) 3. Replace existing task strings wherever they appear (using `task_replacements`)
You can combine all of them: `new_task` sets the default, while `episode_tasks` and You can combine all of them: `new_task` sets the default, while `episode_tasks` and
`task_replacements` override specific episodes, with `episode_tasks` taking precedence. `task_replacements` override specific episodes, with `episode_tasks` taking precedence.
The dataset is modified in-place, updating only the task-related files: The dataset is modified in-place, updating only the task-related files:
tests/datasets/test_dataset_tools.py
+3 -1
View File
@@ -1177,7 +1177,9 @@ def test_modify_tasks_default_task_and_replacements(sample_dataset):
def test_modify_tasks_no_task_specified(sample_dataset): def test_modify_tasks_no_task_specified(sample_dataset):
"""Test error when no task is specified.""" """Test error when no task is specified."""
with pytest.raises(ValueError, match="Must specify at least one of new_task, episode_tasks, or task_replacements"): with pytest.raises(
ValueError, match="Must specify at least one of new_task, episode_tasks, or task_replacements"
):
modify_tasks(sample_dataset) modify_tasks(sample_dataset)