From 4cbd0e14be6140b0d71eacfa41451a33e73c2361 Mon Sep 17 00:00:00 2001 From: Nautics889 Date: Sun, 10 Sep 2023 17:31:54 +0300 Subject: [PATCH] chore: type hint issue * (chore): update return type for `df_type` function --- pandasai/helpers/df_info.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pandasai/helpers/df_info.py b/pandasai/helpers/df_info.py index ac3de91f3..a22dd5b90 100644 --- a/pandasai/helpers/df_info.py +++ b/pandasai/helpers/df_info.py @@ -11,7 +11,7 @@ DataFrameType = Union[pd.DataFrame, str] -def df_type(df: DataFrameType) -> str: +def df_type(df: DataFrameType) -> Union[str, None]: """ Returns the type of the dataframe.