make consistent for both branches

2025-06-24 22:15:04 -04:00
parent 687e1ddb53
commit af9a3d933b
1 changed files with 1 additions and 0 deletions
--- a/python/lammps/pylammps.py
+++ b/python/lammps/pylammps.py
@ -132,6 +132,7 @@ class AtomList(object):
    if index not in self._loaded:
      if self.dimensions == 2:
        atom = Atom2D(self._pylmp, index)
+        self._loaded[index] = atom
      else:
        atom = Atom(self._pylmp, index)
        self._loaded[index] = atom