From e10d89d8c4ae911229a35aef607059ff08df360b Mon Sep 17 00:00:00 2001 From: Richard Berger Date: Fri, 10 Sep 2021 12:20:02 -0400 Subject: [PATCH] Add omega field to numpy_wrapper detection --- python/lammps/numpy_wrapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/lammps/numpy_wrapper.py b/python/lammps/numpy_wrapper.py index 2f0f74594e..b29503a271 100644 --- a/python/lammps/numpy_wrapper.py +++ b/python/lammps/numpy_wrapper.py @@ -92,7 +92,7 @@ class numpy_wrapper: if dim == LAMMPS_AUTODETECT: if dtype in (LAMMPS_INT_2D, LAMMPS_DOUBLE_2D, LAMMPS_INT64_2D): # TODO add other fields - if name in ("x", "v", "f", "angmom", "torque", "csforce", "vforce"): + if name in ("x", "v", "f", "omega", "angmom", "torque", "csforce", "vforce"): dim = 3 else: dim = 2