From ea5009e767b4e05dd906739800dc28892bc81eda Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Thu, 28 Nov 2024 11:38:12 +0100 Subject: [PATCH] Remove commented code --- tests/test_control_robot.py | 12 ------------ 1 file changed, 12 deletions(-) diff --git a/tests/test_control_robot.py b/tests/test_control_robot.py index c51ca972f..0ba737a8f 100644 --- a/tests/test_control_robot.py +++ b/tests/test_control_robot.py @@ -295,24 +295,12 @@ def test_resume_record(tmpdir, request, robot_type, mock): dataset = record(**record_kwargs) assert len(dataset) == 1, f"`dataset` should contain 1 frame, not {len(dataset)}" - # init_dataset_return_value = {} - - # def wrapped_init_dataset(*args, **kwargs): - # nonlocal init_dataset_return_value - # init_dataset_return_value = init_dataset(*args, **kwargs) - # return init_dataset_return_value - - # with patch("lerobot.scripts.control_robot.init_dataset", wraps=wrapped_init_dataset): - with pytest.raises(FileExistsError): # Dataset already exists, but resume=False by default record(**record_kwargs) dataset = record(**record_kwargs, resume=True) assert len(dataset) == 2, f"`dataset` should contain 2 frames, not {len(dataset)}" - # assert ( - # init_dataset_return_value["num_episodes"] == 2 - # ), "`init_dataset` should load the previous episode" @pytest.mark.parametrize("robot_type, mock", [("koch", True)])