Use pydantic shims (#85)

joostlek · Jul 28, 2023 · 1964cf8 · 1964cf8
1 parent d4d97d8
commit 1964cf8
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 13 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/src/python_opensky/models.py b/src/python_opensky/models.py
@@ -3,7 +3,10 @@
 
 from dataclasses import dataclass
 
-from pydantic import BaseModel, Field
+try:
+    from pydantic.v1 import BaseModel, Field
+except ImportError:  # pragma: no cover
+    from pydantic import BaseModel, Field  # type: ignore[assignment] # pragma: no cover
 
 from .const import AircraftCategory, PositionSource
 from .exceptions import OpenSkyCoordinateError