From 2c2bcc7c55283a999d604b74e45a6088e628973d Mon Sep 17 00:00:00 2001 From: Maximilian Linhoff Date: Wed, 23 Nov 2022 17:41:31 +0100 Subject: [PATCH] Adjust chunksize for tests --- ctapipe/tools/tests/test_apply_models.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ctapipe/tools/tests/test_apply_models.py b/ctapipe/tools/tests/test_apply_models.py index c19ba618990..2735f91a531 100644 --- a/ctapipe/tools/tests/test_apply_models.py +++ b/ctapipe/tools/tests/test_apply_models.py @@ -26,6 +26,7 @@ def test_apply_energy_regressor( f"--output={output_path}", f"--energy-regressor={energy_regressor_path}", "--StereoMeanCombiner.weights=konrad", + "--chunk-size=5", # small chunksize so we test multiple chunks for the test file ], raises=True, ) @@ -86,6 +87,7 @@ def test_apply_both( f"--energy-regressor={energy_regressor_path}", f"--particle-classifier={particle_classifier_path}", "--StereoMeanCombiner.weights=konrad", + "--chunk-size=5", # small chunksize so we test multiple chunks for the test file ], ) assert ret == 0