From 3e7de1ed91e805eafd37ac890d30243778db50ad Mon Sep 17 00:00:00 2001 From: Hassan Kibirige Date: Fri, 17 May 2024 20:36:01 +0300 Subject: [PATCH] TYP: Add type annotations to function --- plotnine/coords/coord_cartesian.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/plotnine/coords/coord_cartesian.py b/plotnine/coords/coord_cartesian.py index c50a9b588..2b28cfa48 100644 --- 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)