tx258/lammps-gran-kokkos
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

simplify

Browse Source
This commit is contained in:
Axel Kohlmeyer
2025-06-26 14:42:13 -04:00
parent f763420c53
commit 4a58be05fa
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/lammps/pylammps.py
View File

@ -132,10 +132,9 @@ class AtomList(object):
if index not in self._loaded: if index not in self._loaded:
if self.dimensions == 2: if self.dimensions == 2:
atom = Atom2D(self._pylmp, index) atom = Atom2D(self._pylmp, index)
self._loaded[index] = atom
else: else:
atom = Atom(self._pylmp, index) atom = Atom(self._pylmp, index)
self._loaded[index] = atom self._loaded[index] = atom
return self._loaded[index] return self._loaded[index]
def __len__(self): def __len__(self):