Fix evoformer paths in some old tests

tests/test_import_weights.py

@@ -65,7 +65,7 @@ class TestImportWeights(unittest.TestCase):
                         )
                     ][1].transpose(-1, -2)
                 ),
-                model.evoformer.blocks[1].outer_product_mean.linear_1.weight,
+                model.evoformer.blocks[1].core.outer_product_mean.linear_1.weight,
             ),
         ]
 

tests/test_outer_product_mean.py

@@ -81,7 +81,7 @@ class TestOuterProductMean(unittest.TestCase):
 
         model = compare_utils.get_global_pretrained_openfold()
         out_repro = (
-            model.evoformer.blocks[0]
+            model.evoformer.blocks[0].core
             .outer_product_mean(
                 torch.as_tensor(msa_act).cuda(),
                 chunk_size=4,

tests/test_pair_transition.py

@@ -76,7 +76,7 @@ class TestPairTransition(unittest.TestCase):
 
         model = compare_utils.get_global_pretrained_openfold()
         out_repro = (
-            model.evoformer.blocks[0]
+            model.evoformer.blocks[0].core
             .pair_transition(
                 torch.as_tensor(pair_act, dtype=torch.float32).cuda(),
                 chunk_size=4,