RunasSudo/scipy-yli
1
Fork 0
Code Issues 1 Pull Requests Releases Activity

Fix bug when nan_policy='raise'

Browse Source
This commit is contained in:
RunasSudo 2022-10-20 20:59:06 +11:00
parent b31ae6686f
commit bae93b2c8e
Signed by: RunasSudo
GPG Key ID: 7234E476BF21C61A
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
yli/utils.py
View File

@ -31,6 +31,7 @@ def check_nan(df, nan_policy):
if nan_policy == 'raise': if nan_policy == 'raise':
if pd.isna(df).any(axis=None): if pd.isna(df).any(axis=None):
raise ValueError('NaN in input, pass nan_policy="warn" or "omit" to ignore') raise ValueError('NaN in input, pass nan_policy="warn" or "omit" to ignore')
return df
elif nan_policy == 'warn': elif nan_policy == 'warn':
df_cleaned = df.dropna() df_cleaned = df.dropna()
if len(df_cleaned) < len(df): if len(df_cleaned) < len(df):