diff --git a/scripts/export.py b/scripts/export.py index e01200d..ede7691 100644 --- a/scripts/export.py +++ b/scripts/export.py @@ -23,6 +23,7 @@ RUN_ID_FIELD = "USER28" OUTLIER_TYPE_CODES = { + None: None, "Low price": "1", "High price": "2", "Low price per square foot": "3", @@ -112,6 +113,8 @@ for field in not_null_fields: assert flag_df[flag_df[field].isnull()].empty, f"{field} contains nulls" + logger.info(f"Got {num_flags} sales with flags") + for field in [OUTLIER_REASON1_FIELD, OUTLIER_REASON2_FIELD, OUTLIER_REASON3_FIELD]: invalid_values = flag_df[~flag_df[field].isin(OUTLIER_TYPE_CODES.values())] assert invalid_values.empty, f"{field} contains invalid codes"