Fix another inference-time data parser bug

c58f3cd8 · Gustaf Ahdritz · d65b76a2 · c58f3cd8
Commit c58f3cd8 authored Oct 20, 2021 by Gustaf Ahdritz
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/openfold/data/data_transforms.py
+++ b/openfold/data/data_transforms.py
@@ -1132,7 +1132,7 @@ def random_crop_to_size(
        num_templates_crop_size = num_templates
    n = seq_length - num_res_crop_size
-    if protein["use_clamped_fape"] == 1.:
+    if "use_clamped_fape" in protein and protein["use_clamped_fape"] == 1.:
        right_anchor = n
    else:
        x = _randint(0, n)