From ccd2a8e7607107aff214fc922fe5190be93f3ee9 Mon Sep 17 00:00:00 2001 From: Micah Woodard Date: Thu, 13 Aug 2026 10:32:31 -0700 Subject: [PATCH] updates stage names in metrics to fix validation errors --- .../src/aind_behavior_dynamic_foraging_curricula/metrics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/workspace/aind_behavior_dynamic_foraging_curricula/src/aind_behavior_dynamic_foraging_curricula/metrics.py b/workspace/aind_behavior_dynamic_foraging_curricula/src/aind_behavior_dynamic_foraging_curricula/metrics.py index 9010a4a..bf1c40a 100644 --- a/workspace/aind_behavior_dynamic_foraging_curricula/src/aind_behavior_dynamic_foraging_curricula/metrics.py +++ b/workspace/aind_behavior_dynamic_foraging_curricula/src/aind_behavior_dynamic_foraging_curricula/metrics.py @@ -7,7 +7,7 @@ from aind_behavior_dynamic_foraging.task_logic.utils.calculate_foraging_efficiency import calculate_foraging_efficiency from pydantic import BeforeValidator, Field -STAGE_NAMES = Literal["stage_1_warmup", "stage_1", "stage_2", "stage_3", "final", "graduated"] +STAGE_NAMES = Literal["STAGE_1_WARMUP", "STAGE_1", "STAGE_2", "STAGE_3", "STAGE_FINAL", "GRADUATED"] logger = logging.getLogger(__name__)