fix a bug

sail-sg · Nov 24, 2023 · 888facf · 888facf
1 parent dbb3a52
commit 888facf
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/d4ft/integral/gto/tensorization.py b/d4ft/integral/gto/tensorization.py
@@ -154,9 +154,6 @@ def tensorize(
     N_abcd = Ns[abcd_idx].prod(-1) * counts_abcd_i
     abcd = jnp.einsum("k,k,k,k,k,k->k", t_abcd, N_abcd, *coeffs_abcd)
     cgto_abcd = jax.ops.segment_sum(abcd, cgto_seg_id, n_segs)
-    print(cgto_abcd.shape)
-    print(cgto_seg_id.shape)
-    print(n_segs)
     return cgto_abcd
 
   return tensorize