chore: type hint issue

* (chore): update return type for `df_type` function
Sinaptik-AI · Sep 10, 2023 · 4cbd0e1 · 4cbd0e1
1 parent 4ad7296
commit 4cbd0e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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.