Remove redundancy in data_transforms

b32bfeec · Gustaf Ahdritz · 03bb003a · b32bfeec
Commit b32bfeec authored Nov 23, 2021 by Gustaf Ahdritz
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

openfold/data/data_transforms.py openfold/data/data_transforms.py +4 -4

No files found.
--- a/openfold/data/data_transforms.py
+++ b/openfold/data/data_transforms.py
@@ -1131,12 +1131,12 @@ def random_crop_to_size(
        templates_select_indices = torch.randperm(
            num_templates, device=protein["seq_length"].device, generator=g
        )
+    else:
+        templates_crop_start = 0
+
    num_templates_crop_size = min(
        num_templates - templates_crop_start, max_templates
    )
-    else:
-        templates_crop_start = 0
-        num_templates_crop_size = num_templates

    n = seq_length - num_res_crop_size
    if "use_clamped_fape" in protein and protein["use_clamped_fape"] == 1.: