From 091ab1ff5fb4d13854cf6a7c41ec531e1d9d3e67 Mon Sep 17 00:00:00 2001 From: Justas Dauparas Date: Mon, 6 May 2024 22:40:06 -0700 Subject: [PATCH] add .long for group_idx --- sc_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sc_utils.py b/sc_utils.py index 9a688f4..5f30779 100644 --- a/sc_utils.py +++ b/sc_utils.py @@ -208,12 +208,12 @@ def make_torsion_features(feature_dict, repack_everything=True): S_af2, torch.tensor(restype_rigid_group_default_frame, device=device), ) - + xyz14_noised = feats.frames_and_literature_positions_to_atom14_pos( pred_frames, S_af2, torch.tensor(restype_rigid_group_default_frame, device=device), - torch.tensor(restype_atom14_to_rigid_group, device=device), + torch.tensor(restype_atom14_to_rigid_group, device=device).long(), torch.tensor(restype_atom14_mask, device=device), torch.tensor(restype_atom14_rigid_group_positions, device=device), )