From 0f5d56b6177a0a1b85c562809b214e81ef05923b Mon Sep 17 00:00:00 2001
From: Konstantin <konstantin.gulin@gmail.com>
Date: Wed, 28 Jun 2023 14:39:22 +0000
Subject: [PATCH] Reduce IC test footprint

---
 tests/integration/auto/test_cli_run.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/integration/auto/test_cli_run.py b/tests/integration/auto/test_cli_run.py
index d947a89e..e7b3c23f 100644
--- a/tests/integration/auto/test_cli_run.py
+++ b/tests/integration/auto/test_cli_run.py
@@ -157,7 +157,7 @@
                 "--data",
                 "imagenette",
                 "--train-kwargs",
-                "{'max_train_steps': 5,'max_eval_steps': 5}",
+                "{'max_train_steps': 5,'max_eval_steps': 5, 'train_batch_size': 4, 'test_batch_size': 4}",  # noqa: E501
             ],
             False,
         ),