aboutsummaryrefslogtreecommitdiff
path: root/Python/pywarpx/_libwarpx.py
diff options
context:
space:
mode:
Diffstat (limited to 'Python/pywarpx/_libwarpx.py')
-rwxr-xr-xPython/pywarpx/_libwarpx.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/Python/pywarpx/_libwarpx.py b/Python/pywarpx/_libwarpx.py
index 9f588f806..5ffb74360 100755
--- a/Python/pywarpx/_libwarpx.py
+++ b/Python/pywarpx/_libwarpx.py
@@ -311,8 +311,10 @@ class LibWarpX():
dimensions = {'x' : 0, 'z' : 1}
elif self.geometry_dim == '1d':
dimensions = {'z' : 0}
+ elif self.geometry_dim == 'rz':
+ dimensions = {'r': 0, 'z': 1}
else:
- raise NotImplementedError("RZ is not supported for particle scraping.")
+ raise RuntimeError(f"Unknown simulation geometry: {self.geometry_dim}")
if boundary != 'eb':
boundary_parts = boundary.split("_")