Fix: return correct renaming mask.

PiperOrigin-RevId: 558140222
Change-Id: I7fb69e8a7c9ce95a26d0816fb8c1c02cba68f660
This commit is contained in:
DeepMind
2023-08-18 07:26:14 -07:00
committed by Copybara-Service
parent 82ef354ca3
commit bbe8a844e0

View File

@@ -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,