explictly mark overrides

Author: amorison

src/stagpy/_caching.py

@@ -6,6 +6,7 @@
 from collections import deque
 from dataclasses import dataclass
 from functools import cached_property
+from typing import override
 
 from . import phyvars, stagyyparsers
 
@@ -105,12 +106,15 @@ def _info(self) -> StepSnapInfo:
             isnap_max=isnap,
         )
 
+    @override
     def istep(self, *, isnap: int) -> int | None:
         return self._info.snap_to_step.get(isnap)
 
+    @override
     def isnap(self, *, istep: int) -> int | None:
         return self._info.step_to_snap.get(istep)
 
+    @override
     def len_snap(self) -> int:
         return self._info.isnap_max + 1
 
@@ -139,9 +143,11 @@ def isnap_max(self) -> int:
                 imax = max(int(match.group(2)), imax)
         return imax
 
+    @override
     def len_snap(self) -> int:
         return self.isnap_max + 1
 
+    @override
     def istep(self, *, isnap: int) -> int | None:
         if isnap < 0 or isnap > self.isnap_max:
             return None
@@ -157,6 +163,7 @@ def istep(self, *, isnap: int) -> int | None:
                 self._step_to_snap[istep] = isnap
         return istep
 
+    @override
     def isnap(self, *, istep: int) -> int | None:
         if istep < 0:
             return None

src/stagpy/stagyydata.py

@@ -14,6 +14,7 @@
 from functools import cached_property
 from itertools import zip_longest
 from pathlib import Path
+from typing import override
 
 import numpy as np
 
@@ -258,6 +259,7 @@ def _dtimes(self) -> NDArray[np.float64]:
         midpoints = (self._times[:-1] + self._times[1:]) / 2
         return np.diff(midpoints, prepend=self._times[0], append=self._times[-1])
 
+    @override
     def __getitem__(self, name: str) -> dt.Rprof:
         if name in self._cached_data:
             return self._cached_data[name]
@@ -273,6 +275,7 @@ def __getitem__(self, name: str) -> dt.Rprof:
         return self._cached_data[name]
 
     @property
+    @override
     def stepstr(self) -> str:
         return self.steps.stepstr
 
@@ -282,14 +285,17 @@ def _first_rprofs(self) -> step.RprofsInstant:
         return first_step.rprofs
 
     @property
+    @override
     def centers(self) -> NDArray[np.float64]:
         return self._first_rprofs.centers
 
     @property
+    @override
     def walls(self) -> NDArray[np.float64]:
         return self._first_rprofs.walls
 
     @property
+    @override
     def bounds(self) -> tuple[float, float]:
         return self._first_rprofs.bounds
 
@@ -327,6 +333,7 @@ class Steps:
 
     sdat: StagyyData
 
+    @override
     def __repr__(self) -> str:
         return f"{self.sdat!r}.steps"
 
@@ -421,6 +428,7 @@ class Snaps:
 
     sdat: StagyyData
 
+    @override
     def __repr__(self) -> str:
         return f"{self.sdat!r}.snaps"
 
@@ -624,6 +632,7 @@ def __iter__(self) -> Iterator[Step]:
             elif self._pass(item):
                 yield self.over[item]
 
+    @override
     def __eq__(self, other: object) -> bool:
         if not isinstance(other, abc.Iterable):
             return NotImplemented

src/stagpy/step.py

@@ -12,6 +12,7 @@
 from abc import ABC, abstractmethod
 from dataclasses import dataclass
 from functools import cached_property
+from typing import override
 
 import numpy as np
 
@@ -483,6 +484,7 @@ def _rprofs(self) -> DataFrame:
             )
         return self._data
 
+    @override
     def __getitem__(self, name: str) -> Rprof:
         step = self.step
         if name in self._rprofs.columns:
@@ -509,6 +511,7 @@ def __getitem__(self, name: str) -> Rprof:
         return Rprof(rprof, rad, meta)
 
     @property
+    @override
     def stepstr(self) -> str:
         """String representation of the parent :class:`Step`."""
         return str(self.step.istep)
@@ -618,6 +621,7 @@ def rprofs(self) -> RprofsInstant:
         """Radial profiles available at this time step."""
         return RprofsInstant(self)
 
+    @override
     def __repr__(self) -> str:
         if self.isnap is not None:
             return f"{self.sdat!r}.snaps[{self.isnap}]"