diff --git a/mesa_geo/__init__.py b/mesa_geo/__init__.py index 69cae3c9..8021bd5c 100644 --- a/mesa_geo/__init__.py +++ b/mesa_geo/__init__.py @@ -3,6 +3,7 @@ Core Objects: GeoSpace, GeoAgent """ + import datetime from mesa_geo import visualization diff --git a/mesa_geo/raster_layers.py b/mesa_geo/raster_layers.py index 8bf077e6..09c287cd 100644 --- a/mesa_geo/raster_layers.py +++ b/mesa_geo/raster_layers.py @@ -243,16 +243,15 @@ def attributes(self) -> set[str]: return self._attributes @overload - def __getitem__(self, index: int) -> list[Cell]: - ... + def __getitem__(self, index: int) -> list[Cell]: ... @overload - def __getitem__(self, index: tuple[int | slice, int | slice]) -> Cell | list[Cell]: - ... + def __getitem__( + self, index: tuple[int | slice, int | slice] + ) -> Cell | list[Cell]: ... @overload - def __getitem__(self, index: Sequence[Coordinate]) -> list[Cell]: - ... + def __getitem__(self, index: Sequence[Coordinate]) -> list[Cell]: ... def __getitem__( self, index: int | Sequence[Coordinate] | tuple[int | slice, int | slice]