commit change

mosaicml · Jun 3, 2024 · 79ad163 · 79ad163
1 parent 7bcdc30
commit 79ad163
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/data/test_icl_datasets.py b/tests/data/test_icl_datasets.py
@@ -40,11 +40,11 @@ def run_test(
         if i == 0:
             if eval_drop_last:
                 assert len(
-                    e.dataloader.dataloader
+                    e.dataloader.dataloader,
                 ) == EXPECTED_FIRST_DATALOADER_LEN - 1
             else:
                 assert len(
-                    e.dataloader.dataloader
+                    e.dataloader.dataloader,
                 ) == EXPECTED_FIRST_DATALOADER_LEN
 
         inputs = batch['input_ids'][0]