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

formatting

Browse Source
This commit is contained in:
nemo
2025-11-24 18:52:18 +01:00
parent 83aac1b42e
commit e544562c39
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_cli.py
+4 -4
View File
@@ -27,7 +27,7 @@ def test_peft_training_push_to_hub_works(policy_type, tmp_path):
upload_folder_contents = set() upload_folder_contents = set()
def mock_upload_folder(*args, **kwargs): def mock_upload_folder(*args, **kwargs):
folder_path = kwargs['folder_path'] folder_path = kwargs["folder_path"]
# we include more than is actually uploaded since we ignore {allow,ignore}_patterns of upload_folders() # we include more than is actually uploaded since we ignore {allow,ignore}_patterns of upload_folders()
upload_folder_contents.update(os.listdir(folder_path)) upload_folder_contents.update(os.listdir(folder_path))
return MagicMock() return MagicMock()
@@ -49,9 +49,9 @@ def test_peft_training_push_to_hub_works(policy_type, tmp_path):
] ]
) )
assert 'adapter_model.safetensors' in upload_folder_contents assert "adapter_model.safetensors" in upload_folder_contents
assert 'config.json' in upload_folder_contents assert "config.json" in upload_folder_contents
assert 'adapter_config.json' in upload_folder_contents assert "adapter_config.json" in upload_folder_contents
@pytest.mark.parametrize("policy_type", ["smolvla"]) @pytest.mark.parametrize("policy_type", ["smolvla"])