diff --git a/aliby/tile/tiler.py b/aliby/tile/tiler.py index 6305f51d72764a717d211bbdfbfccd216eab2a11..3920a50f6b584411dd33bea47dff540b4ce293be 100644 --- a/aliby/tile/tiler.py +++ b/aliby/tile/tiler.py @@ -47,7 +47,7 @@ class Trap: self.half_size = size // 2 self.max_size = max_size - def at_time(self, tp): + def at_time(self, tp: int) -> t.List[int]: """ Return trap centre at time tp by applying drifts @@ -62,7 +62,7 @@ class Trap: """ drifts = self.parent.drifts trap_centre = self.centre - np.sum(drifts[: tp + 1], axis=0) - return trap_centre + return list(trap_centre.astype(int)) def as_tile(self, tp): """