diff --git a/xbatcher/tests/test_generators.py b/xbatcher/tests/test_generators.py index b0939dc..3665ba7 100644 --- a/xbatcher/tests/test_generators.py +++ b/xbatcher/tests/test_generators.py @@ -224,6 +224,7 @@ def test_batch_3d_1d_input(sample_ds_3d, input_size): 'concat', [ True, + False, ], ) def test_batch_3d_1d_input_batch_dims(sample_ds_3d, concat):