Hot fix to compute validation loss example test (#200)

Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
2024-05-20 18:30:11 +02:00
parent 9b62c25f6c
commit c4da689171
3 changed files with 50 additions and 6 deletions
--- a/tests/test_examples.py
+++ b/tests/test_examples.py
@@ -111,7 +111,7 @@ def test_examples_2_through_4():
                '# pretrained_policy_path = Path("outputs/train/example_pusht_diffusion")',
                'pretrained_policy_path = Path("outputs/train/example_pusht_diffusion")',
            ),
-            ('split="train[24342:]"', 'split="train[-1:]"'),
+            ('split=f"train[{first_val_frame_index}:]"', 'split="train[30:]"'),
            ("num_workers=4", "num_workers=0"),
            ('device = torch.device("cuda")', 'device = torch.device("cpu")'),
            ("batch_size=64", "batch_size=1"),