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":