From bae93b2c8ef62b413978fa23d9764c45adfebb6c Mon Sep 17 00:00:00 2001 From: RunasSudo Date: Thu, 20 Oct 2022 20:59:06 +1100 Subject: [PATCH] Fix bug when nan_policy='raise' --- yli/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/yli/utils.py b/yli/utils.py index a19acc5..ce95f0a 100644 --- a/yli/utils.py +++ b/yli/utils.py @@ -31,6 +31,7 @@ def check_nan(df, nan_policy): if nan_policy == 'raise': if pd.isna(df).any(axis=None): raise ValueError('NaN in input, pass nan_policy="warn" or "omit" to ignore') + return df elif nan_policy == 'warn': df_cleaned = df.dropna() if len(df_cleaned) < len(df):