Unverified Commit ff1065ba authored by Gao, Xiang's avatar Gao, Xiang Committed by GitHub
Browse files

More cleanup (#176)

parent 56eba454
...@@ -195,7 +195,7 @@ class AEVComputer(torch.nn.Module): ...@@ -195,7 +195,7 @@ class AEVComputer(torch.nn.Module):
cutoff radius are valid. The returned indices stores the source of data cutoff radius are valid. The returned indices stores the source of data
before sorting. before sorting.
""" """
max_cutoff = max([self.Rcr, self.Rca]) max_cutoff = max(self.Rcr, self.Rca)
species_, distances, vec = self.neighborlist(species, coordinates, species_, distances, vec = self.neighborlist(species, coordinates,
max_cutoff) max_cutoff)
radial_terms = self._radial_subaev_terms(distances) radial_terms = self._radial_subaev_terms(distances)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment