Merge pull request #65 from halvarsu:remove_atoms_from_protocol

PiperOrigin-RevId: 556750621 Change-Id: I5d084f9823cc9d119630c1b9a8519286a369bc9a
google-deepmind · Aug 14, 2023 · 6d17339 · 6d17339
2 parents e29145a + 5602a0c
commit 6d17339
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/ferminet/hamiltonian.py b/ferminet/hamiltonian.py
@@ -50,7 +50,6 @@ class MakeLocalEnergy(Protocol):
   def __call__(
       self,
       f: networks.FermiNetLike,
-      atoms: jnp.ndarray,
       charges: jnp.ndarray,
       nspins: Sequence[int],
       use_scan: bool = False,
@@ -62,7 +61,6 @@ def __call__(
     Args:
       f: Callable which evaluates the sign and log of the magnitude of the
         wavefunction.
-      atoms: atomic positions.
       charges: nuclear charges.
       nspins: Number of particles of each spin.
       use_scan: Whether to use a `lax.scan` for computing the laplacian.