Make SerializedPipelineGraph compatible with pydantic v2.

lsst · Jul 24, 2023 · 7841684 · 7841684
1 parent 60f68d0
commit 7841684
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/lsst/pipe/base/pipeline_graph/io.py b/python/lsst/pipe/base/pipeline_graph/io.py
@@ -36,6 +36,7 @@
 import networkx
 import pydantic
 from lsst.daf.butler import DatasetType, DimensionConfig, DimensionGraph, DimensionUniverse
+from lsst.daf.butler._compat import _BaseModelCompat
 
 from .. import automatic_connection_constants as acc
 from ._dataset_types import DatasetTypeNode
@@ -78,7 +79,7 @@ def expect_not_none(value: _U | None, msg: str) -> _U:
     return value
 
 
-class SerializedEdge(pydantic.BaseModel):
+class SerializedEdge(_BaseModelCompat):
     """Struct used to represent a serialized `Edge` in a `PipelineGraph`.
 
     All `ReadEdge` and `WriteEdge` state not included here is instead