diff --git a/src/aind_data_schema/core/model.py b/src/aind_data_schema/core/model.py index 5de0259c..fe09320f 100644 --- a/src/aind_data_schema/core/model.py +++ b/src/aind_data_schema/core/model.py @@ -62,7 +62,7 @@ class Model(AindCoreModel): license: str = Field(..., title="License") developer_full_name: Optional[List[str]] = Field(default=None, title="Name of developer") developer_institution: Optional[Organization.ONE_OF] = Field(default=None, title="Institute where developed") - modality: Modality.ONE_OF = Field(..., title="Modality") + modality: List[Modality.ONE_OF] = Field(..., title="Modality") architecture: ModelArchitecture = Field(..., title="Model architecture") intended_use: str = Field(..., title="Intended model use", description="Semantic description of intended use") limitations: Optional[str] = Field(default=None, title="Model limitations") diff --git a/tests/test_model.py b/tests/test_model.py index d57b2cf3..b93e7d8b 100644 --- a/tests/test_model.py +++ b/tests/test_model.py @@ -28,7 +28,7 @@ def test_constructors(self): license="CC-BY-4.0", developer_full_name=["Joe Schmoe"], developer_institution=Organization.AIND, - modality=Modality.SPIM, + modality=[Modality.SPIM], pretrained_source_url="url pretrained weights are from", architecture=ModelArchitecture( backbone=ModelBackbone.RESNET,