More cleanup (#176)

ff1065ba · Gao, Xiang · GitHub · 56eba454 · ff1065ba
Unverified Commit ff1065ba authored Mar 04, 2019 by Gao, Xiang Committed by GitHub Mar 04, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchani/aev.py torchani/aev.py +1 -1

No files found.
--- a/torchani/aev.py
+++ b/torchani/aev.py
@@ -195,7 +195,7 @@ class AEVComputer(torch.nn.Module):
        cutoff radius are valid. The returned indices stores the source of data
        before sorting.
        """
-        max_cutoff = max([self.Rcr, self.Rca])
+        max_cutoff = max(self.Rcr, self.Rca)
        species_, distances, vec = self.neighborlist(species, coordinates,
                                                     max_cutoff)
        radial_terms = self._radial_subaev_terms(distances)