Minor fix to TaxonomyEntry type & initial data loading (#138)

climatepolicyradar · Jun 12, 2023 · b97de0b · b97de0b
1 parent a56d999
commit b97de0b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/app/core/ingestion/metadata.py b/app/core/ingestion/metadata.py
@@ -1,4 +1,4 @@
-from typing import Any, Mapping, Sequence, Union, Optional
+from typing import Any, Mapping, Sequence, Union
 
 from pydantic.dataclasses import dataclass
 from pydantic.config import ConfigDict, Extra
@@ -13,7 +13,7 @@ class TaxonomyEntry:
 
     allow_blanks: bool
     allowed_values: Sequence[str]
-    allow_any: Optional[bool] = False
+    allow_any: bool = False
 
 
 Taxonomy = Mapping[str, TaxonomyEntry]

diff --git a/app/data_migrations/populate_taxonomy.py b/app/data_migrations/populate_taxonomy.py
@@ -24,6 +24,7 @@ def populate_org_taxonomy(
         )
         db.add(org)
         db.flush()
+        db.commit()
 
     metadata_org = (
         db.query(MetadataOrganisation)
@@ -46,6 +47,7 @@ def populate_org_taxonomy(
             )
         )
         db.flush()
+        db.commit()
 
 
 def populate_taxonomy(db: Session) -> None: