Add stage_alias property to TaskParameters child

AllenNeuralDynamics · May 17, 2024 · a219677 · a219677
1 parent baf2847
commit a219677
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/DataSchemas/aind_behavior_services/task_logic/__init__.py b/src/DataSchemas/aind_behavior_services/task_logic/__init__.py
@@ -7,6 +7,7 @@
 
 class TaskParameters(curriculum_task.TaskParameters):
     rng_seed: Optional[float] = Field(default=None, description="Seed of the random number generator")
+    stage_alias: Optional[str] = Field(default=None, description="Alias name used for the task stage")
 
 
 class AindBehaviorTaskLogicModel(curriculum_task.Task):