Use pydantic shims

joostlek · Jul 21, 2023 · a98f95f · a98f95f
1 parent 60080dc
commit a98f95f
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/python_opensky/models.py b/src/python_opensky/models.py
@@ -3,20 +3,23 @@
 
 from dataclasses import dataclass
 
-from pydantic import BaseModel, Field
+try:
+    from pydantic.v1 import BaseModel, Field
+except ImportError:
+    from pydantic import BaseModel, Field
 
 from .const import AircraftCategory, PositionSource
 from .exceptions import OpenSkyCoordinateError
 
 
-class StatesResponse(BaseModel):
+class StatesResponse(BaseModel):  # type: ignore[misc]
     """Represents the states response."""
 
     states: list[StateVector] = Field(...)
     time: int = Field(...)
 
 
-class StateVector(BaseModel):
+class StateVector(BaseModel):  # type: ignore[misc]
     """Represents the state of a vehicle at a particular time.
 
     Attributes