squash into 7786f46

DeepRank · Sep 22, 2023 · 131ea78 · 131ea78
1 parent d039645
commit 131ea78
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deeprank2/molstruct/aminoacid.py b/deeprank2/molstruct/aminoacid.py
@@ -1,6 +1,7 @@
 from enum import Enum
 
 import numpy as np
+from numpy.typing import NDArray
 
 
 class Polarity(Enum):
@@ -108,7 +109,7 @@ def hydrogen_bond_acceptors(self) -> int:
         return self._hydrogen_bond_acceptors
 
     @property
-    def onehot(self) -> np.ndarray:
+    def onehot(self) -> NDArray:
         if self._index is None:
             raise ValueError(
                 f"Amino acid {self._name} index is not set, thus no onehot can be computed."