From 549c5b9949bb46f06e9330ca46b0c3018763f5c3 Mon Sep 17 00:00:00 2001
From: Martin Kozlovsky <martin.kozlovsky@luxonis.com>
Date: Thu, 30 May 2024 18:36:45 +0200
Subject: [PATCH] decreased batch size for tests

---
 tests/integration/test_sanity.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/integration/test_sanity.py b/tests/integration/test_sanity.py
index 8b6f872b..efb3ded7 100644
--- a/tests/integration/test_sanity.py
+++ b/tests/integration/test_sanity.py
@@ -22,6 +22,8 @@ def test_sanity(config_file):
         "1",
         "trainer.callbacks",
         "[]",
+        "trainer.batch_size",
+        "1",
     ]
     result = subprocess.run(
         ["luxonis_train", "train", "--config", f"configs/{config_file}", *opts],
@@ -80,6 +82,8 @@ def test_tuner():
             "[]",
             "tuner.n_trials",
             "4",
+            "trainer.batch_size",
+            "1",
         ],
     )
     assert result.returncode == 0