Update impute_missing_values to handle SAS table NaN values

sassoftware · Dec 19, 2023 · 1c032e2 · 1c032e2
1 parent 9e1e2ae
commit 1c032e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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