fix: fix type hint from float to Number

GispoCoding · Sep 18, 2024 · 5d93c8c · 5d93c8c
1 parent d36dad2
commit 5d93c8c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/eis_toolkit/exploratory_analyses/descriptive_statistics.py b/eis_toolkit/exploratory_analyses/descriptive_statistics.py
@@ -1,3 +1,5 @@
+from numbers import Number
+
 import geopandas as gpd
 import numpy as np
 import pandas as pd
@@ -10,7 +12,9 @@
 from eis_toolkit.exceptions import InvalidColumnException, InvalidRasterBandException
 
 
-def _descriptive_statistics(data: Union[rasterio.io.DatasetReader, pd.DataFrame, gpd.GeoDataFrame]) -> Dict[str, float]:
+def _descriptive_statistics(
+    data: Union[rasterio.io.DatasetReader, pd.DataFrame, gpd.GeoDataFrame]
+) -> Dict[str, Number]:
     statistics = DescrStatsW(data)
     min = np.min(data)
     max = np.max(data)
@@ -69,7 +73,7 @@ def descriptive_statistics_dataframe(
 
 
 @beartype
-def descriptive_statistics_raster(input_data: rasterio.io.DatasetReader, band: int = 1) -> Dict[str, float]:
+def descriptive_statistics_raster(input_data: rasterio.io.DatasetReader, band: int = 1) -> Dict[str, Number]:
     """Compute descriptive statistics from raster data.
 
     Computes the following statistics: