Skip to content

Commit 1c032e2

Browse files
committed
Update impute_missing_values to handle SAS table NaN values
1 parent 9e1e2ae commit 1c032e2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sasctl/pzmm/write_score_code.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -721,10 +721,10 @@ def impute_missing_values(data):
721721
"""
722722
impute_values = \\\n + {"var1": 0, "var2": "", "var3": 125.3}
723723
"""
724-
cls.score_code += f"\n{'':4}return data.fillna(impute_values)\n"
724+
cls.score_code += f"\n{'':4}return data.replace(' .', np.nan).fillna(impute_values).apply(pd.to_numeric, errors='ignore')\n"
725725
"""
726726
727-
return data.fillna(impute_values)
727+
return data.replace(' .', np.nan).fillna(impute_values).apply(pd.to_numeric, errors='ignore')
728728
"""
729729

730730
# TODO: Needs unit test

0 commit comments

Comments
 (0)