Skip to content

Commit

Permalink
fix for DataFrameNaFunctions return type hints (#1195)
Browse files Browse the repository at this point in the history
Co-authored-by: t.charts <[email protected]>
  • Loading branch information
sfc-gh-aalam and TedCha authored Jan 5, 2024
1 parent 098da14 commit 366bbf5
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/snowflake/snowpark/dataframe_na_functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,15 +60,15 @@ 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(
self,
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
Expand Down Expand Up @@ -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.
Expand Down Expand Up @@ -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.
Expand Down

0 comments on commit 366bbf5

Please sign in to comment.