From bbe8a844e09ba24c439c067affca9b4b0d637026 Mon Sep 17 00:00:00 2001 From: DeepMind Date: Fri, 18 Aug 2023 07:26:14 -0700 Subject: [PATCH] Fix: return correct renaming mask. PiperOrigin-RevId: 558140222 Change-Id: I7fb69e8a7c9ce95a26d0816fb8c1c02cba68f660 --- alphafold/model/folding_multimer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/alphafold/model/folding_multimer.py b/alphafold/model/folding_multimer.py index b565a0d..1c0d689 100644 --- a/alphafold/model/folding_multimer.py +++ b/alphafold/model/folding_multimer.py @@ -775,7 +775,7 @@ def compute_atom14_gt( gt_mask = (1. - use_alt) * gt_mask + use_alt * alt_gt_mask gt_positions = (1. - use_alt) * gt_positions + use_alt * alt_gt_positions - return gt_positions, alt_gt_mask, alt_naming_is_better + return gt_positions, gt_mask, alt_naming_is_better def backbone_loss(gt_rigid: geometry.Rigid3Array,