Skip to content

Commit 6a8712a

Browse files
committed
chore: add type hint
1 parent 3a417d7 commit 6a8712a

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

src/libecalc/presentation/yaml/domain/time_series_expression.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import numpy as np
2+
from numpy.typing import NDArray
23

34
from libecalc.common.variables import ExpressionEvaluator
45
from libecalc.dto.utils.validators import convert_expression
@@ -53,7 +54,7 @@ def get_evaluated_expressions(self) -> list[float]:
5354
arr = np.atleast_1d(arr[0]) # Flattens (1, N) to (N,)
5455
return arr.tolist()
5556

56-
def get_condition_mask(self):
57+
def get_condition_mask(self) -> NDArray[np.int_] | None:
5758
if self._condition is None:
5859
return None
5960
mask = self.expression_evaluator.evaluate(expression=self._condition)

0 commit comments

Comments
 (0)