Fix datatypes in PDB parsing

e0c7d445 · Gustaf Ahdritz · 699f23fd · e0c7d445
Commit e0c7d445 authored Nov 30, 2021 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

openfold/data/data_pipeline.py openfold/data/data_pipeline.py +2 -2

No files found.
--- a/openfold/data/data_pipeline.py
+++ b/openfold/data/data_pipeline.py
@@ -147,8 +147,8 @@ def make_protein_features(
    all_atom_positions = protein_object.atom_positions
    all_atom_mask = protein_object.atom_mask

-    pdb_feats["all_atom_positions"] = all_atom_positions
-    pdb_feats["all_atom_mask"] = all_atom_mask
+    pdb_feats["all_atom_positions"] = all_atom_positions.astype(np.float32)
+    pdb_feats["all_atom_mask"] = all_atom_mask.astype(np.float32)

    pdb_feats["resolution"] = np.array([0.]).astype(np.float32)
    pdb_feats["is_distillation"] = np.array(