diff --git a/python/lsst/pipe/base/pipeline_graph/io.py b/python/lsst/pipe/base/pipeline_graph/io.py index 33c9ef85..060591d4 100644 --- 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