fix typing

districtr · Feb 3, 2025 · 7054b6a · 7054b6a
1 parent c420ecc
commit 7054b6a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/backend/app/main.py b/backend/app/main.py
@@ -345,7 +345,8 @@ async def get_unassigned_geoids(
     results = session.execute(
         stmt, {"doc_uuid": document_id, "exclude_ids": exclude_ids}
     ).fetchall()
-    return {"features": [row[0] for row in results]}
+    # returns a list of multipolygons of bboxes
+    return {"features": results}
 
 
 @app.get("/api/document/{document_id}/evaluation/{summary_stat}")

diff --git a/backend/app/models.py b/backend/app/models.py
@@ -16,7 +16,7 @@
 )
 from sqlalchemy.types import ARRAY, TEXT
 from sqlalchemy import Float
-from pydantic_geojson import FeatureModel
+from pydantic_geojson.multi_polygon import MultiPolygonModel
 from app.constants import DOCUMENT_SCHEMA
 from enum import Enum
 from typing import Any
@@ -197,7 +197,7 @@ class AssignedGEOIDS(GEOIDS):
 
 
 class UnassignedBboxGeoJSONs(BaseModel):
-    features: list[FeatureModel]
+    features: list[MultiPolygonModel]
 
 
 class ShatterResult(BaseModel):