Add csforce field to numpy_wrapper detection

2022-02-03 20:05:18 -05:00
parent efcb402fdc
commit 2627e404b0
1 changed files with 1 additions and 1 deletions
--- 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", "x0", "omega", "angmom", "torque", "vforce", "vest"):
+        if name in ("x", "v", "f", "x0","omega", "angmom", "torque", "csforce", "vforce", "vest"):
          dim = 3
        elif name == "smd_data_9":
          dim = 9