From 4ac8c4a364224f1a4c6b9d13611aa002e277f79d Mon Sep 17 00:00:00 2001 From: Dan Birman Date: Wed, 6 Nov 2024 19:02:38 -0800 Subject: [PATCH] fix: remove references to non-existant ModalityModel --- src/aind_data_schema/core/quality_control.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/aind_data_schema/core/quality_control.py b/src/aind_data_schema/core/quality_control.py index 15c2d0337..46604a48f 100644 --- 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),