Update to new enum nancodes

Author: dshemetov

safegraph_patterns/delphi_safegraph_patterns/process.py

@@ -6,7 +6,7 @@
 import numpy as np
 import pandas as pd
 
-from delphi_utils import create_export_csv, GeoMapper, NAN_CODES
+from delphi_utils import create_export_csv, GeoMapper, Nans
 
 INCIDENCE_BASE = 100000
 
@@ -127,15 +127,15 @@ def aggregate(df, metric, geo_res):
 def add_nancodes(df):
     """Add nan codes to the df."""
     # Values are not missing unless found to be null
-    df["missing_val"] = NAN_CODES["Not Missing"]
+    df["missing_val"] = Nans.NOT_MISSING
 
     # These values aren't reported, so are always missing
-    df["missing_se"] = NAN_CODES["Not Applicable"]
-    df["missing_sample_size"] = NAN_CODES["Not Applicable"]
+    df["missing_se"] = Nans.NOT_APPLICABLE
+    df["missing_sample_size"] = Nans.NOT_APPLICABLE
 
     # Mark any remaining nans with unknown
     remaining_nans_mask = (df["missing_val"] == 0) & df["val"].isnull()
-    df.loc[remaining_nans_mask, "missing_val"] = NAN_CODES["Unknown"]
+    df.loc[remaining_nans_mask, "missing_val"] = Nans.UNKNOWN
 
     return df