Use args when calling self.aev_computer (#118)

* Use args when calling self.aev_computer. [ref aiqm/torchani#649] * Trivial change to try to get CI to run --------- Co-authored-by: Peter Eastman <[email protected]>
openmm · Jul 10, 2024 · 05d30d4 · 05d30d4
1 parent d15cb91
commit 05d30d4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/pytorch/OptimizedTorchANI.py b/src/pytorch/OptimizedTorchANI.py
@@ -49,8 +49,8 @@ def forward(self, species_coordinates: Tuple[Tensor, Tensor],
                 pbc: Optional[Tensor] = None) -> SpeciesEnergies:
 
         species_coordinates = self.species_converter(species_coordinates)
-        species_aevs = self.aev_computer(species_coordinates, cell=cell, pbc=pbc)
+        species_aevs = self.aev_computer(species_coordinates, cell, pbc)
         species_energies = self.neural_networks(species_aevs)
         species_energies = self.energy_shifter(species_energies)
 
-        return species_energies
+        return species_energies