fixed alignments when permutation is turned off

058fc74f · Geoffrey Yu · a9d65037 · 058fc74f
Commit 058fc74f authored Oct 03, 2023 by Geoffrey Yu
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 3 deletions

openfold/utils/loss.py openfold/utils/loss.py +1 -3

No files found.
--- a/openfold/utils/loss.py
+++ b/openfold/utils/loss.py
@@ -2196,9 +2196,7 @@ class AlphaFoldMultimerLoss(AlphaFoldLoss):
            
        else:
            per_asym_residue_index = AlphaFoldMultimerLoss.get_per_asym_residue_index(feature)
-            labels = AlphaFoldMultimerLoss.split_ground_truth_labels(ground_truth,
-                                                                 REQUIRED_FEATURES=["all_atom_mask","all_atom_positions"])
-            best_align = list(enumerate(range(len(labels))))
+            best_align = list(enumerate(range(len(per_asym_residue_index))))
        
        return best_align, per_asym_residue_index