From 1b009663954110a10b59c4a8d8d071ffd4e503a7 Mon Sep 17 00:00:00 2001
From: ivarflakstad <69173633+ivarflakstad@users.noreply.github.com>
Date: Wed, 14 May 2025 14:12:04 +0200
Subject: [PATCH] Fix auto batch size finder test (#38125)

Ensure --auto_find_batch_size is the last test arg so indexing is correct
---
 tests/trainer/test_trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py
index 6708c484eb..f687d2d88e 100644
--- a/tests/trainer/test_trainer.py
+++ b/tests/trainer/test_trainer.py
@@ -3300,8 +3300,8 @@ class TrainerIntegrationTest(TestCasePlus, TrainerIntegrationCommon):
                 --learning_rate 2e-5
                 --num_train_epochs 1
                 --output_dir {tmpdir}
-                --auto_find_batch_size 0
                 --report_to none
+                --auto_find_batch_size 0
                 """.split()
             with self.assertRaises(RuntimeError):
                 with patch.object(sys, "argv", testargs):