set correct size for zero acceptance in grid and vector configuration (…

…#716)
atcollab · Dec 4, 2023 · 2f72fc2 · 2f72fc2
1 parent d95aa5c
commit 2f72fc2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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)