Enum value get consistently sorted

mysociety · Sep 4, 2023 · a3b8e93 · a3b8e93
1 parent 23605c6
commit a3b8e93
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/data_common/dataset/table_management.py b/src/data_common/dataset/table_management.py
@@ -3,6 +3,7 @@
 
 import pandas as pd
 from pandas.io.json import build_table_schema
+
 from data_common.db import duck_query
 
 
@@ -80,6 +81,9 @@ def enhance_field(
  field["constraints"]["enum"] = enum_value
  if isinstance(enum_value, EnumPlaceholder):
  field["constraints"]["enum"] = enum_value.process(col)
+ if isinstance(field["constraints"]["enum"], list):
+ # sort the enum values
+ field["constraints"]["enum"] = sorted(field["constraints"]["enum"])
  return field
 
  @classmethod
@@ -105,7 +109,6 @@ def get_table_schema(
 def update_table_schema(
  path: Path, existing_schema: SchemaValidator | None
 ) -> SchemaValidator:
-
  if path.suffix == ".csv":
  df = pd.read_csv(path)
  elif path.suffix == ".parquet":