diff --git a/src/primaite/environment/observations.py b/src/primaite/environment/observations.py index b19bd29f..81ddaaf5 100644 --- a/src/primaite/environment/observations.py +++ b/src/primaite/environment/observations.py @@ -29,7 +29,7 @@ class AbstractObservationComponent(ABC): self.env: "Primaite" = env self.space: spaces.Space self.current_observation: np.ndarray # type might be too restrictive? - self.structure: list[str] + self.structure: List[str] return NotImplemented @abstractmethod