Remove redundant condition

1eb4fcdd · Gustaf Ahdritz · GitHub · 576174f0 · 1eb4fcdd
Unverified Commit 1eb4fcdd authored May 07, 2022 by Gustaf Ahdritz Committed by GitHub May 07, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

openfold/model/structure_module.py openfold/model/structure_module.py +1 -2

No files found.
--- a/openfold/model/structure_module.py
+++ b/openfold/model/structure_module.py
@@ -665,8 +665,7 @@ class StructureModule(nn.Module):

            outputs.append(preds)

-            if i < (self.no_blocks - 1):
-                rigids = rigids.stop_rot_gradient()
+            rigids = rigids.stop_rot_gradient()

        outputs = dict_multimap(torch.stack, outputs)
        outputs["single"] = s