From 366bbf54ca0b8a985d2c8331114903344599cbb3 Mon Sep 17 00:00:00 2001 From: Afroz Alam Date: Fri, 5 Jan 2024 22:34:33 +0000 Subject: [PATCH] fix for DataFrameNaFunctions return type hints (#1195) Co-authored-by: t.charts <60014013+TedCha@users.noreply.github.com> --- src/snowflake/snowpark/dataframe_na_functions.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/snowflake/snowpark/dataframe_na_functions.py b/src/snowflake/snowpark/dataframe_na_functions.py index aecc5070fba..11cc7081c56 100644 --- a/src/snowflake/snowpark/dataframe_na_functions.py +++ b/src/snowflake/snowpark/dataframe_na_functions.py @@ -60,7 +60,7 @@ def _is_value_type_matching_for_na_function( class DataFrameNaFunctions: """Provides functions for handling missing values in a :class:`DataFrame`.""" - def __init__(self, df: "snowflake.snowpark.dataframe.DataFrame") -> None: + def __init__(self, df: "snowflake.snowpark.DataFrame") -> None: self._df = df def drop( @@ -68,7 +68,7 @@ def drop( how: str = "any", thresh: Optional[int] = None, subset: Optional[Union[str, Iterable[str]]] = None, - ) -> "snowflake.snowpark.dataframe.DataFrame": + ) -> "snowflake.snowpark.DataFrame": """ Returns a new DataFrame that excludes all rows containing fewer than a specified number of non-null and non-NaN values in the specified @@ -219,7 +219,7 @@ def fill( self, value: Union[LiteralType, Dict[str, LiteralType]], subset: Optional[Union[str, Iterable[str]]] = None, - ) -> "snowflake.snowpark.dataframe.DataFrame": + ) -> "snowflake.snowpark.DataFrame": """ Returns a new DataFrame that replaces all null and NaN values in the specified columns with the values provided. @@ -400,7 +400,7 @@ def replace( ], value: Optional[Union[LiteralType, Iterable[LiteralType]]] = None, subset: Optional[Iterable[str]] = None, - ) -> "snowflake.snowpark.dataframe.DataFrame": + ) -> "snowflake.snowpark.DataFrame": """ Returns a new DataFrame that replaces values in the specified columns.