I want MolecularHamiltonian.local energy(ansatz) to be available on an already constructed…

I want MolecularHamiltonian.local energy(ansatz) to be available on an already constructed…: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). In a session like local energy fn = hamil.local energy(ansatz.apply); energy, stats = local energy fn(rng, params, phys…