Update pydantic_definitions.py

int-brain-lab · Aug 31, 2024 · 6226a4d · 6226a4d
1 parent d6c6b95
commit 6226a4d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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