diff --git a/nwb_linkml/src/nwb_linkml/includes/hdmf.py b/nwb_linkml/src/nwb_linkml/includes/hdmf.py index addc32c..77b484c 100644 --- a/nwb_linkml/src/nwb_linkml/includes/hdmf.py +++ b/nwb_linkml/src/nwb_linkml/includes/hdmf.py @@ -16,9 +16,9 @@ from typing import ( ) import numpy as np +import pandas as pd from linkml.generators.pydanticgen.template import Import, Imports, ObjectImport from numpydantic import NDArray, Shape -import pandas as pd from pydantic import ( BaseModel, ConfigDict, @@ -236,7 +236,7 @@ class DynamicTableMixin(BaseModel): return model @model_validator(mode="after") - def cast_extra_columns(self): + def cast_extra_columns(self) -> "DynamicTableMixin": """ If extra columns are passed as just lists or arrays, cast to VectorData before we resolve targets for VectorData and VectorIndex pairs.