Merge pull request #1143 from AllenNeuralDynamics/patch-modality-model

fix: remove references to non-existant ModalityModel
AllenNeuralDynamics · Nov 7, 2024 · 23f7b45 · 23f7b45
2 parents f6ef0c3 + 4ac8c4a
commit 23f7b45
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/aind_data_schema/core/quality_control.py b/src/aind_data_schema/core/quality_control.py
@@ -4,7 +4,7 @@
 from enum import Enum
 from typing import Any, List, Literal, Optional, Union
 
-from aind_data_schema_models.modalities import Modality, ModalityModel
+from aind_data_schema_models.modalities import Modality
 from pydantic import BaseModel, Field, SkipValidation, field_validator, model_validator
 
 from aind_data_schema.base import AindCoreModel, AindModel, AwareDatetimeWithDefault
@@ -180,7 +180,7 @@ class QualityControl(AindCoreModel):
 
     def status(
         self,
-        modality: Union[ModalityModel, List[ModalityModel], None] = None,
+        modality: Union[Modality.ONE_OF, List[Modality.ONE_OF], None] = None,
         stage: Union[Stage, List[Stage], None] = None,
         tag: Union[str, List[str], None] = None,
         date: datetime = datetime.now(tz=timezone.utc),