Reduce IC test footprint

neuralmagic · Jun 28, 2023 · 116b15a · 116b15a
1 parent 0c461f8
commit 116b15a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,
         ),