Skip to content

Commit e48fbb9

Browse files
committed
Added check for label column in neighbors since it is expected down the road for that process
1 parent e382085 commit e48fbb9

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

exatomic/algorithms/neighbors.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -140,6 +140,8 @@ def sorter(group, source_atom_idxs):
140140
s = group[['atom0', 'atom1']].stack()
141141
return s[~s.isin(source_atom_idxs)].reset_index()
142142

143+
if "label" not in uni.atom.columns:
144+
uni.atom['label'] = uni.atom.get_atom_labels()
143145
dct = defaultdict(list)
144146
grps = uni.atom.groupby("frame")
145147
ntot = len(grps)

0 commit comments

Comments
 (0)