fix: make model modality into list

AllenNeuralDynamics · Dec 13, 2024 · a27ddc0 · a27ddc0
1 parent 5d9ae35
commit a27ddc0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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,