Merge pull request #355 from dingquanyu/fix-non-permutation-codes

fixed the creation of best_align when permutation is turned off

Merge pull request #355 from dingquanyu/fix-non-permutation-codes
fixed the creation of best_align when permutation is turned off
377f854c · Christina Floristean · GitHub · 0ca66146 · 058fc74f · 377f854c
Unverified Commit 377f854c authored Oct 04, 2023 by Christina Floristean Committed by GitHub Oct 04, 2023
Show 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