From 1c032e2e634e8bbbe6f5ad24f2eb16ae98883bf8 Mon Sep 17 00:00:00 2001 From: djm21 Date: Tue, 19 Dec 2023 11:21:56 -0500 Subject: [PATCH] Update impute_missing_values to handle SAS table NaN values --- src/sasctl/pzmm/write_score_code.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/sasctl/pzmm/write_score_code.py b/src/sasctl/pzmm/write_score_code.py index 056097c6..fac02a06 100644 --- a/src/sasctl/pzmm/write_score_code.py +++ b/src/sasctl/pzmm/write_score_code.py @@ -721,10 +721,10 @@ def impute_missing_values(data): """ impute_values = \\\n + {"var1": 0, "var2": "", "var3": 125.3} """ - cls.score_code += f"\n{'':4}return data.fillna(impute_values)\n" + cls.score_code += f"\n{'':4}return data.replace(' .', np.nan).fillna(impute_values).apply(pd.to_numeric, errors='ignore')\n" """ - return data.fillna(impute_values) + return data.replace(' .', np.nan).fillna(impute_values).apply(pd.to_numeric, errors='ignore') """ # TODO: Needs unit test