Revert earlier change

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(),
-        ca_xyz=batch["template_all_atom_positions"][..., ca, :].float(),
-        c_xyz=batch["template_all_atom_positions"][..., c, :].float(),
+        n_xyz=batch["template_all_atom_positions"][..., n, :],
+        ca_xyz=batch["template_all_atom_positions"][..., ca, :],
+        c_xyz=batch["template_all_atom_positions"][..., c, :],
         eps=eps,
     )
     points = rigids.get_trans()[..., None, :, :]