Merge pull request #465 from DeepRank/463_fix_preprocess

fix: cast `hse` feature to float64
DeepRank · Aug 23, 2023 · 4091a9e · 4091a9e
2 parents b152a4b + 07b17d3
commit 4091a9e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deeprankcore/features/exposure.py b/deeprankcore/features/exposure.py
@@ -72,6 +72,6 @@ def add_features( # pylint: disable=unused-argument
             hse_key = (residue.chain.id, (" ", residue.number, space_if_none(residue.insertion_code)))
 
             if hse_key in hse:
-                node.features[Nfeat.HSE] = hse[hse_key]
+                node.features[Nfeat.HSE] = np.array(hse[hse_key], dtype=np.float64)
             else:
-                node.features[Nfeat.HSE] = np.array((0, 0, 0))
+                node.features[Nfeat.HSE] = np.array((0, 0, 0), dtype=np.float64)