add type annotation

matthiasdemuzere · Jul 12, 2022 · 4b7f6f0 · 4b7f6f0
1 parent 673e0e4
commit 4b7f6f0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/w2w/w2w.py b/w2w/w2w.py
@@ -381,7 +381,10 @@ def check_lcz_integrity(info: Info, LCZ_BAND: int) -> None:
     lcz.rio.to_raster(info.src_file_clean, dtype=np.int32)
 
 
-def using_kdtree(data, kpoints):
+def using_kdtree(
+    data: pd.DataFrame,
+    kpoints: int,
+) -> Tuple[NDArray[np.int_], NDArray[np.int_]]:
 
     '''
     Extract nearest kpoints to each point given a list of them.