Remove prints from test

sirmarcel · Oct 31, 2022 · 42c59e2 · 42c59e2
1 parent 2cb74a8
commit 42c59e2
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/tests/test_cg.py b/tests/test_cg.py
@@ -22,16 +22,12 @@ def test_U_matrix_shape():
 def test_U_matrix_mace_compare():
     irreps_in = Irreps("1x0e + 1x1o + 1x2e")
     irreps_out = Irreps("1x0e + 1x1o")
-    print("initialized irreps jax")
     irreps_in_mace = o3.Irreps("1x0e + 1x1o + 1x2e")
     irreps_out_mace = o3.Irreps("1x0e + 1x1o")
-    print("initialized irreps e3nn")
     u_matrix = cg.U_matrix_real(
         irreps_in=irreps_in, irreps_out=irreps_out, correlation=3
     )[-1]
-    print("initialized matrix jax")
     u_matrix_mace = mace_cg.U_matrix_real(
         irreps_in=irreps_in_mace, irreps_out=irreps_out_mace, correlation=3
     )[-1]
-    print("initialized matrix e3nn")
     assert jnp.allclose(u_matrix, u_matrix_mace.cpu().detach().numpy())