Fix for spatial cropping indexing

8e522aca · Christina Floristean · 6713a264 · 8e522aca
Commit 8e522aca authored Aug 29, 2023 by Christina Floristean
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/openfold/data/data_transforms_multimer.py
+++ b/openfold/data/data_transforms_multimer.py
@@ -347,7 +347,7 @@ def get_spatial_crop_idx(protein, crop_size, interface_threshold, generator):
        return get_contiguous_crop_idx(protein, crop_size, generator)
    target_res_idx = randint(lower=0,
-                             upper=interface_residues.shape[-1],
+                             upper=interface_residues.shape[-1] - 1,
                             generator=generator,
                             device=positions.device)