From 6226a4d26d3f041e0bd53efbf1d509b536e9fccb Mon Sep 17 00:00:00 2001 From: Florian Rau Date: Sat, 31 Aug 2024 20:16:01 +0100 Subject: [PATCH] Update pydantic_definitions.py --- iblrig/pydantic_definitions.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/iblrig/pydantic_definitions.py b/iblrig/pydantic_definitions.py index fbd58d397..7fcb4a199 100644 --- a/iblrig/pydantic_definitions.py +++ b/iblrig/pydantic_definitions.py @@ -246,6 +246,7 @@ def preallocate_dataframe(cls, n_rows: int) -> pd.DataFrame: class TrialDataChoiceWorld(TrialDataModel): """Definition of Trial Data for ChoiceWorldSession""" + contrast: Annotated[float, Ge(0.0), Le(1.0)] position: float quiescent_period: Annotated[float, Ge(0.0)] @@ -271,6 +272,7 @@ class TrialDataChoiceWorld(TrialDataModel): class TrialDataActiveChoiceWorld(TrialDataChoiceWorld): """Definition of Trial Data for ActiveChoiceWorldSession""" + response_side: Annotated[int, Ge(-1), Le(1)] response_time: Annotated[float, Ge(0.0)] trial_correct: bool