fix mcgn test

NLESC-JCER · Nov 29, 2023 · 4c2a354 · 4c2a354
1 parent 6085953
commit 4c2a354
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/wavefunction/test_slater_mgcn_graph_jastrow.py b/tests/wavefunction/test_slater_mgcn_graph_jastrow.py
@@ -156,7 +156,7 @@ def test_kinetic_energy(self):
         assert torch.allclose(eauto.data, ejac.data, rtol=1e-4, atol=1e-4)
 
     def test_gradients_wf(self):
-        grads = self.wf.gradients_jacobi_no_backflow(self.pos, sum_grad=False).squeeze()
+        grads = self.wf.gradients_jacobi(self.pos, sum_grad=False).squeeze()
         grad_auto = self.wf.gradients_autograd(self.pos)
 
         assert torch.allclose(grads.sum(), grad_auto.sum())
@@ -166,7 +166,7 @@ def test_gradients_wf(self):
         assert torch.allclose(grads, grad_auto)
 
     def test_gradients_pdf(self):
-        grads_pdf = self.wf.gradients_jacobi_no_backflow(self.pos, pdf=True)
+        grads_pdf = self.wf.gradients_jacobi(self.pos, pdf=True)
         grads_auto = self.wf.gradients_autograd(self.pos, pdf=True)
 
         assert torch.allclose(grads_pdf.sum(), grads_auto.sum())