Skip to content
Snippets Groups Projects
Commit 546386bb authored by paradoxe-tech's avatar paradoxe-tech
Browse files

fixed critical error (pocket -> protein)

parent 1d5c9853
No related branches found
No related tags found
No related merge requests found
......@@ -36,7 +36,7 @@ def interaction(receptor_path:str, ligand_path:str) -> torch.Tensor:
protein_dict=torchify_dict(pocket_dict),
ligand_dict=torchify_dict(ligand_dict),
residue_dict=torchify_dict(residue_dict),
seq=''.join(residue_dict['seq']),
seq=''.join(protein.to_dict_residue()['seq']),
full_seq_index=torch.tensor(full_seq_index),
r10_index=torch.tensor(r10_index)
)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment