Commit e0c7d445 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Fix datatypes in PDB parsing

parent 699f23fd
...@@ -147,8 +147,8 @@ def make_protein_features( ...@@ -147,8 +147,8 @@ def make_protein_features(
all_atom_positions = protein_object.atom_positions all_atom_positions = protein_object.atom_positions
all_atom_mask = protein_object.atom_mask all_atom_mask = protein_object.atom_mask
pdb_feats["all_atom_positions"] = all_atom_positions pdb_feats["all_atom_positions"] = all_atom_positions.astype(np.float32)
pdb_feats["all_atom_mask"] = all_atom_mask pdb_feats["all_atom_mask"] = all_atom_mask.astype(np.float32)
pdb_feats["resolution"] = np.array([0.]).astype(np.float32) pdb_feats["resolution"] = np.array([0.]).astype(np.float32)
pdb_feats["is_distillation"] = np.array( pdb_feats["is_distillation"] = np.array(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment