1212import sympy
1313from scipy .sparse import issparse , vstack
1414from collections .abc import Sequence
15+ from typing import Any
1516
1617import pybamm
1718
@@ -114,7 +115,7 @@ def get_children_domains(self, children: Sequence[pybamm.Symbol]):
114115
115116 return domains
116117
117- def _concatenation_evaluate (self , children_eval : list [npt .NDArray ]):
118+ def _concatenation_evaluate (self , children_eval : list [npt .NDArray [ Any ] ]):
118119 """See :meth:`Concatenation._concatenation_evaluate()`."""
119120 if len (children_eval ) == 0 :
120121 return np .array ([])
@@ -124,8 +125,8 @@ def _concatenation_evaluate(self, children_eval: list[npt.NDArray]):
124125 def evaluate (
125126 self ,
126127 t : float | None = None ,
127- y : npt .NDArray | None = None ,
128- y_dot : npt .NDArray | None = None ,
128+ y : npt .NDArray [ np . float64 ] | None = None ,
129+ y_dot : npt .NDArray [ np . float64 ] | None = None ,
129130 inputs : dict | str | None = None ,
130131 ):
131132 """See :meth:`pybamm.Symbol.evaluate()`."""
@@ -369,7 +370,7 @@ def create_slices(self, node: pybamm.Symbol) -> defaultdict:
369370 start = end
370371 return slices
371372
372- def _concatenation_evaluate (self , children_eval : list [npt .NDArray ]):
373+ def _concatenation_evaluate (self , children_eval : list [npt .NDArray [ Any ] ]):
373374 """See :meth:`Concatenation._concatenation_evaluate()`."""
374375 # preallocate vector
375376 vector = np .empty ((self ._size , 1 ))
0 commit comments