Fix test_examples

tests/test_examples.py

@@ -29,13 +29,14 @@ def test_examples_3_and_2():
     with open(path, "r") as file:
         file_contents = file.read()
 
-    # Do less steps, use CPU, and don't complicate things with dataloader workers.
+    # Do less steps, use smaller batch, use CPU, and don't complicate things with dataloader workers.
     file_contents = _find_and_replace(
         file_contents,
         [
             ("training_steps = 5000", "training_steps = 1"),
             ("num_workers=4", "num_workers=0"),
             ('device = torch.device("cuda")', 'device = torch.device("cpu")'),
+            ("batch_size=cfg.batch_size", "batch_size=1"),
         ],
     )