TYP: Add type annotations to function

has2k1 · May 17, 2024 · 3e7de1e · 3e7de1e
1 parent aec1837
commit 3e7de1e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/plotnine/coords/coord_cartesian.py b/plotnine/coords/coord_cartesian.py
@@ -52,10 +52,10 @@ def transform(
     ) -> pd.DataFrame:
         from mizani.bounds import squish_infinite
 
-        def squish_infinite_x(col):
+        def squish_infinite_x(col: FloatSeries) -> FloatArray:
             return squish_infinite(col, range=panel_params.x.range)
 
-        def squish_infinite_y(col):
+        def squish_infinite_y(col: FloatSeries) -> FloatArray:
             return squish_infinite(col, range=panel_params.y.range)
 
         return transform_position(data, squish_infinite_x, squish_infinite_y)