simplify

2025-06-26 14:42:13 -04:00
parent f763420c53
commit 4a58be05fa
1 changed files with 1 additions and 2 deletions
--- a/python/lammps/pylammps.py
+++ b/python/lammps/pylammps.py
@ -132,10 +132,9 @@ 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
+      self._loaded[index] = atom
    return self._loaded[index]

  def __len__(self):