Fix tiny but consequential bug in excited states

PiperOrigin-RevId: 577488269 Change-Id: I4c1d3601ac150e6f59f821ce3a50f2659798d82a
google-deepmind · Nov 24, 2023 · e9f8c64 · e9f8c64
1 parent 3fb9e5e
commit e9f8c64
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ferminet/networks.py b/ferminet/networks.py
@@ -1476,7 +1476,7 @@ def apply(
           jnp.reshape(orbital, (options.states, -1) + orbital.shape[1:])
           for orbital in orbitals
       ]
-      return batch_logdet_matmul(*orbitals)
+      return batch_logdet_matmul(orbitals)
     else:
       return network_blocks.logdet_matmul(orbitals)