Add omega field to numpy_wrapper detection

This commit is contained in:
Richard Berger
2021-09-10 12:20:02 -04:00
parent d09851e695
commit e10d89d8c4

View File

@ -92,7 +92,7 @@ class numpy_wrapper:
if dim == LAMMPS_AUTODETECT: if dim == LAMMPS_AUTODETECT:
if dtype in (LAMMPS_INT_2D, LAMMPS_DOUBLE_2D, LAMMPS_INT64_2D): if dtype in (LAMMPS_INT_2D, LAMMPS_DOUBLE_2D, LAMMPS_INT64_2D):
# TODO add other fields # 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 dim = 3
else: else:
dim = 2 dim = 2