fix type annotations

Author: amorison

src/stagpy/stagyydata.py

@@ -164,6 +164,7 @@ def _tseries(self) -> DataFrame:
         return self._data
 
     def __getitem__(self, name: str) -> dt.Tseries:
+        series: NDArray[np.float64]
         if name in self._tseries.columns:
             series = self._tseries[name].to_numpy(dtype=np.float64)
             time = self.time

src/stagpy/step.py

@@ -485,6 +485,7 @@ def _rprofs(self) -> DataFrame:
 
     def __getitem__(self, name: str) -> Rprof:
         step = self.step
+        rprof: NDArray[np.float64]
         if name in self._rprofs.columns:
             rprof = self._rprofs[name].to_numpy(dtype=np.float64)
             rad = self.centers