From c29f5f359540bfffd5c48cbd5124ac9f12ce8d03 Mon Sep 17 00:00:00 2001 From: Cadene Date: Tue, 23 Apr 2024 11:46:07 +0000 Subject: [PATCH] nit --- tests/scripts/save_dataset_to_safetensors.py | 6 +++--- tests/test_datasets.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/scripts/save_dataset_to_safetensors.py b/tests/scripts/save_dataset_to_safetensors.py index 787d746a..4f0875e2 100644 --- a/tests/scripts/save_dataset_to_safetensors.py +++ b/tests/scripts/save_dataset_to_safetensors.py @@ -51,17 +51,17 @@ def save_dataset_to_safetensors(output_dir, dataset_id="pusht"): # TODO(rcadene): Enable testing on second and last episode # We currently cant because our test dataset only contains the first episode - # save 2 first frames of second episode + # # save 2 first frames of second episode # i = dataset.episode_data_index["from"][1].item() # save_file(dataset[i], data_dir / f"frame_{i}.safetensors") # save_file(dataset[i+1], data_dir / f"frame_{i+1}.safetensors") - # save 2 last frames of second episode + # # save 2 last frames of second episode # i = dataset.episode_data_index["to"][1].item() # save_file(dataset[i-2], data_dir / f"frame_{i-2}.safetensors") # save_file(dataset[i-1], data_dir / f"frame_{i-1}.safetensors") - # save 2 last frames of last episode + # # save 2 last frames of last episode # i = dataset.episode_data_index["to"][-1].item() # save_file(dataset[i-2], data_dir / f"frame_{i-2}.safetensors") # save_file(dataset[i-1], data_dir / f"frame_{i-1}.safetensors") diff --git a/tests/test_datasets.py b/tests/test_datasets.py index 48f81abd..ec459c58 100644 --- a/tests/test_datasets.py +++ b/tests/test_datasets.py @@ -282,17 +282,17 @@ def test_backward_compatibility(): # TODO(rcadene): Enable testing on second and last episode # We currently cant because our test dataset only contains the first episode - # test 2 first frames of second episode + # # test 2 first frames of second episode # i = dataset.episode_data_index["from"][1].item() # load_and_compare(i) # load_and_compare(i+1) - # test 2 last frames of second episode + # #test 2 last frames of second episode # i = dataset.episode_data_index["to"][1].item() # load_and_compare(i-2) # load_and_compare(i-1) - # test 2 last frames of last episode + # # test 2 last frames of last episode # i = dataset.episode_data_index["to"][-1].item() # load_and_compare(i-2) # load_and_compare(i-1)