Revert earlier change

913903e0 · Gustaf Ahdritz · e56b5976 · 913903e0
Commit 913903e0 authored Aug 18, 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, :].float(),
+        n_xyz=batch["template_all_atom_positions"][..., n, :],
-        ca_xyz=batch["template_all_atom_positions"][..., ca, :].float(),
+        ca_xyz=batch["template_all_atom_positions"][..., ca, :],
-        c_xyz=batch["template_all_atom_positions"][..., c, :].float(),
+        c_xyz=batch["template_all_atom_positions"][..., c, :],
        eps=eps,
    )
    points = rigids.get_trans()[..., None, :, :]