fix interpolate

NLESC-JCER · Oct 26, 2023 · 77aaddc · 77aaddc
1 parent 6a4123c
commit 77aaddc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qmctorch/utils/interpolate.py b/qmctorch/utils/interpolate.py
@@ -170,7 +170,7 @@ def func(x):
             bas = self.wf.ao.norm_cst * self.wf.ao.bas_coeffs * bas
             ao = torch.zeros(nbatch, self.wf.ao.nelec,
                              self.wf.ao.norb, device=self.wf.ao.device)
-            print(ao.shape, self.wf.ao.index_ctr.shape, bas.shape)
+            bas = bas.tile(1,self.wf.ao.nelec,1)
             ao.index_add_(2, self.wf.ao.index_ctr, bas)
             return ao