From f7d2c9a1f00b234d62728e12af43dff653f2ac69 Mon Sep 17 00:00:00 2001 From: Dani Bodor Date: Fri, 22 Sep 2023 18:45:40 +0200 Subject: [PATCH] squash into 7786f469 --- deeprank2/molstruct/aminoacid.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/deeprank2/molstruct/aminoacid.py b/deeprank2/molstruct/aminoacid.py index 88c615a87..189b5a564 100644 --- 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."