Add sample_Upsi to infidelity driver

netket · Oct 24, 2023 · c6356c6 · c6356c6
1 parent 3f3ccb2
commit c6356c6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/netket_fidelity/driver/infidelity_optimizer.py b/netket_fidelity/driver/infidelity_optimizer.py
@@ -39,6 +39,7 @@ def __init__(
         U_dagger=None,
         preconditioner: PreconditionerT = identity_preconditioner,
         is_unitary=False,
+        sample_Upsi=False, 
         cv_coeff=-0.5,
     ):
         r"""
@@ -134,6 +135,7 @@ def __init__(
             U_dagger=U_dagger,
             is_unitary=is_unitary,
             cv_coeff=cv_coeff,
+            sample_Upsi=sample_Upsi, 
         )
 
     def _forward_and_backward(self):