Skip to content

GitLab

Commit 4a215372 authored by Geoffrey Yu's avatar Geoffrey Yu
Browse files

fixed index error in cropping

parent dc5d43d7
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
openfold/data/data_transforms_multimer.py
View file @ 4a215372
...@@ -401,8 +401,9 @@ def get_contiguous_crop_idx(protein, crop_size, generator): ...@@ -401,8 +401,9 @@ def get_contiguous_crop_idx(protein, crop_size, generator):
upper=this_len - chain_crop_size + 1, upper=this_len - chain_crop_size + 1,
generator=generator, generator=generator,
device=chain_lens.device) device=chain_lens.device)
cur_asym_id = unique_asym_ids[int(idx)].item()
asym_offset = per_asym_residue_index[int(idx)] print(f"######## 405 cur_asym_id:{cur_asym_id} idx:{idx} and per_asym_residue_index keys: {list(per_asym_residue_index.keys())}")
asym_offset = per_asym_residue_index[int(cur_asym_id)]
crop_idxs.append( crop_idxs.append(
torch.arange(asym_offset + chain_start, asym_offset + chain_start + chain_crop_size) torch.arange(asym_offset + chain_start, asym_offset + chain_start + chain_crop_size)
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment