diff --git a/pyat/at/acceptance/boundary.py b/pyat/at/acceptance/boundary.py index 1279cac21..465018b55 100644 --- a/pyat/at/acceptance/boundary.py +++ b/pyat/at/acceptance/boundary.py @@ -264,7 +264,11 @@ def radial_boundary(mask, grid): msg = ("No particle survived, please check your grid " "or lattice. Acceptance set to [0.0, 0.0].") warnings.warn(AtWarning(msg)) - return numpy.zeros((2, 1)) + cnt = numpy.flip(config.shape) + if len(cnt) == 1: + return numpy.zeros(2) + else: + return numpy.zeros((2, 1)) if config.mode is GridMode.RADIAL: return radial_boundary(mask, grid)