1- from datetime import date , datetime
1+ from datetime import datetime
22from typing import List , Optional , Tuple , Union
3- from ..events .schemas import Event
43
54from pydantic import BaseModel
6- from pydantic import Field , field_validator , model_validator , ConfigDict
5+ from pydantic import Field , ConfigDict
76
87from predicthq .endpoints .schemas import ResultSet
98
10- from typing import Optional , List
119
1210# Python < 3.11 does not have StrEnum in the enum module
1311import sys
@@ -153,22 +151,22 @@ class Point(PhqModel):
153151
154152class Polygon (PhqModel ):
155153 type : Literal ["Polygon" ] = "Polygon"
156- coordinates : list [ list [Position ]]
154+ coordinates : List [ List [Position ]]
157155
158156
159157class MultiPolygon (PhqModel ):
160158 type : Literal ["MultiPolygon" ] = "MultiPolygon"
161- coordinates : list [ list [ list [Position ]]]
159+ coordinates : List [ List [ List [Position ]]]
162160
163161
164162class LineString (PhqModel ):
165163 type : Literal ["LineString" ] = "LineString"
166- coordinates : list [Position ]
164+ coordinates : List [Position ]
167165
168166
169167class MultiLineString (PhqModel ):
170168 type : Literal ["MultiLineString" ] = "MultiLineString"
171- coordinates : list [ list [Position ]]
169+ coordinates : List [ List [Position ]]
172170
173171
174172Geometry = Union [Point , Polygon , MultiPolygon , LineString , MultiLineString ]
@@ -204,9 +202,9 @@ class SavedLocationBase(PhqModel):
204202 location_code : Optional [str ] = None
205203 name : Optional [str ] = None
206204 description : Optional [str ] = None
207- labels : Optional [list [str ]] = None
205+ labels : Optional [List [str ]] = None
208206 geojson : Optional [GeoJson ] = None
209- place_ids : Optional [list [int ]] = None
207+ place_ids : Optional [List [int ]] = None
210208 formatted_address : Optional [str ] = None
211209
212210
@@ -221,11 +219,11 @@ class SavedLocation(SavedLocationBase):
221219 insights_dt : Optional [datetime ] = None
222220 user_id : Optional [str ] = None
223221 subscription_valid_types : Annotated [
224- list [SubscriptionValidType ], Field (default_factory = list )
222+ List [SubscriptionValidType ], Field (default_factory = list )
225223 ]
226224 status : SavedLocationStatus
227- summary_insights : Annotated [list [SummaryInsights ], Field (default_factory = list )]
228- places : Annotated [list [Place ], Field (default_factory = list )]
225+ summary_insights : Annotated [List [SummaryInsights ], Field (default_factory = list )]
226+ places : Annotated [List [Place ], Field (default_factory = list )]
229227
230228
231229class SavedLocationResultSet (ResultSet ):
0 commit comments