From 282a61ce51d1fe619f5bbbf4d1b283fe4c84f8db Mon Sep 17 00:00:00 2001 From: Sarah Hoffmann Date: Tue, 5 Jul 2022 17:28:02 +0200 Subject: [PATCH] add typing information for utils submodule --- nominatim/utils/centroid.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/nominatim/utils/centroid.py b/nominatim/utils/centroid.py index c2bd6192..21823176 100644 --- a/nominatim/utils/centroid.py +++ b/nominatim/utils/centroid.py @@ -7,6 +7,7 @@ """ Functions for computation of centroids. """ +from typing import Tuple, Any from collections.abc import Collection class PointsCentroid: @@ -17,12 +18,12 @@ class PointsCentroid: (i.e. in OSM style). """ - def __init__(self): + def __init__(self) -> None: self.sum_x = 0 self.sum_y = 0 self.count = 0 - def centroid(self): + def centroid(self) -> Tuple[float, float]: """ Return the centroid of all points collected so far. """ if self.count == 0: @@ -32,11 +33,11 @@ class PointsCentroid: float(self.sum_y/self.count)/10000000) - def __len__(self): + def __len__(self) -> int: return self.count - def __iadd__(self, other): + def __iadd__(self, other: Any) -> 'PointsCentroid': if isinstance(other, Collection) and len(other) == 2: if all(isinstance(p, (float, int)) for p in other): x, y = other -- 2.39.5