From 0dfb05f4c1c59d13189222542eb0dc2a7aceb13a Mon Sep 17 00:00:00 2001 From: Saskia de Vries Date: Wed, 8 Jan 2025 15:22:11 -0800 Subject: [PATCH] objects in arena (#1220) * objects in arena * examples * fixed goofs * examples --- examples/bergamo_ophys_session.json | 1 + examples/ephys_session.json | 2 ++ src/aind_data_schema/core/session.py | 2 ++ 3 files changed, 5 insertions(+) diff --git a/examples/bergamo_ophys_session.json b/examples/bergamo_ophys_session.json index ea5c8b9cf..b5910b473 100644 --- a/examples/bergamo_ophys_session.json +++ b/examples/bergamo_ophys_session.json @@ -153,6 +153,7 @@ "stimulus_device_names": [], "speaker_config": null, "light_source_config": [], + "objects_in_arena": null, "output_parameters": {}, "reward_consumed_during_epoch": null, "reward_consumed_unit": "microliter", diff --git a/examples/ephys_session.json b/examples/ephys_session.json index 6d6e42a52..603462f54 100644 --- a/examples/ephys_session.json +++ b/examples/ephys_session.json @@ -326,6 +326,7 @@ "stimulus_device_names": [], "speaker_config": null, "light_source_config": [], + "objects_in_arena": null, "output_parameters": {}, "reward_consumed_during_epoch": null, "reward_consumed_unit": "microliter", @@ -368,6 +369,7 @@ "stimulus_device_names": [], "speaker_config": null, "light_source_config": [], + "objects_in_arena": null, "output_parameters": {}, "reward_consumed_during_epoch": null, "reward_consumed_unit": "microliter", diff --git a/src/aind_data_schema/core/session.py b/src/aind_data_schema/core/session.py index 6c0f1c6c2..5ffcf7e35 100644 --- a/src/aind_data_schema/core/session.py +++ b/src/aind_data_schema/core/session.py @@ -47,6 +47,7 @@ class StimulusModality(str, Enum): """Types of stimulus modalities""" AUDITORY = "Auditory" + FREE_MOVING = "Free moving" OLFACTORY = "Olfactory" OPTOGENETICS = "Optogenetics" NONE = "None" @@ -523,6 +524,7 @@ class StimulusEpoch(AindModel): light_source_config: Optional[List[LIGHT_SOURCE_CONFIGS]] = Field( default=[], title="Light source config", description="Light sources for stimulation" ) + objects_in_arena: Optional[List[str]] = Field(default=None, title="Objects in arena") output_parameters: AindGenericType = Field(default=AindGeneric(), title="Performance metrics") reward_consumed_during_epoch: Optional[Decimal] = Field(default=None, title="Reward consumed during training (uL)") reward_consumed_unit: VolumeUnit = Field(default=VolumeUnit.UL, title="Reward consumed unit")