Remove print statements

Author: gahdritz

tests/test_model.py

@@ -130,6 +130,4 @@ def run_alphafold(batch):
         out_repro = out_repro["sm"]["positions"][-1]
         out_repro = out_repro.squeeze(0)
 
-        print(torch.mean(torch.abs(out_gt - out_repro)))
-        print(torch.max(torch.abs(out_gt - out_repro)))
         self.assertTrue(torch.max(torch.abs(out_gt - out_repro)) < 1e-3)

tests/test_outer_product_mean.py

@@ -90,9 +90,6 @@ def run_opm(msa_act, msa_mask):
             .cpu()
         )
 
-        print(torch.mean(torch.abs(out_gt - out_repro)))
-        print(torch.max(torch.abs(out_gt - out_repro)))
-
         # Even when correct, OPM has large, precision-related errors. It gets
         # a special pass from consts.eps.
         self.assertTrue(torch.max(torch.abs(out_gt - out_repro)) < 5e-4)

tests/test_template.py

@@ -139,8 +139,6 @@ def run_template_pair_stack(pair_act, pair_mask):
             _mask_trans=False,
         ).cpu()
 
-        print(torch.max(torch.abs(out_gt - out_repro)))
-        print(torch.mean(torch.abs(out_gt - out_repro)))
         self.assertTrue(torch.max(torch.abs(out_gt - out_repro)) < consts.eps)
 
 

tests/test_triangular_attention.py

@@ -102,8 +102,6 @@ def run_tri_att(pair_act, pair_mask):
             chunk_size=None,
         ).cpu()
 
-        print(torch.mean(torch.abs(out_gt - out_repro)))
-        print(consts.eps)
         self.assertTrue(torch.mean(torch.abs(out_gt - out_repro)) < consts.eps)
 
     @compare_utils.skip_unless_alphafold_installed()