Increase precision of template operation

6e930a6c · Gustaf Ahdritz · fc7bb8a3 · 6e930a6c
Commit 6e930a6c authored Aug 10, 2022 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

openfold/utils/feats.py openfold/utils/feats.py +3 -3

No files found.
--- a/openfold/utils/feats.py
+++ b/openfold/utils/feats.py
@@ -128,9 +128,9 @@ def build_template_pair_feat(
    n, ca, c = [rc.atom_order[a] for a in ["N", "CA", "C"]]
    rigids = Rigid.make_transform_from_reference(
-        n_xyz=batch["template_all_atom_positions"][..., n, :],
+        n_xyz=batch["template_all_atom_positions"][..., n, :].float(),
-        ca_xyz=batch["template_all_atom_positions"][..., ca, :],
+        ca_xyz=batch["template_all_atom_positions"][..., ca, :].float(),
-        c_xyz=batch["template_all_atom_positions"][..., c, :],
+        c_xyz=batch["template_all_atom_positions"][..., c, :].float(),
        eps=eps,
    )
    points = rigids.get_trans()[..., None, :, :]