From be2024c9c45f0ba042ac05810827f2ba7df23686 Mon Sep 17 00:00:00 2001 From: Micah Woodard Date: Thu, 13 Aug 2026 11:17:16 -0700 Subject: [PATCH 1/2] make lickometers optional --- src/aind_behavior_dynamic_foraging/data_qc/suite.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/aind_behavior_dynamic_foraging/data_qc/suite.py b/src/aind_behavior_dynamic_foraging/data_qc/suite.py index b0602f7..bf65dd9 100644 --- a/src/aind_behavior_dynamic_foraging/data_qc/suite.py +++ b/src/aind_behavior_dynamic_foraging/data_qc/suite.py @@ -68,9 +68,10 @@ def make_qc_runner(dataset: contract.Dataset) -> qc.Runner: "HarpEnvironmentSensor", ) - _runner.add_suite( - qc.harp.HarpLicketySplitTestSuite(dataset["Behavior"]["HarpLickometerRight"]), "HarpLickometerRight" - ) + if rig.harp_lickometer_right is not None: + _runner.add_suite( + qc.harp.HarpLicketySplitTestSuite(dataset["Behavior"]["HarpLickometerRight"]), "HarpLickometerRight" + ) _runner.add_suite( qc.harp.HarpLicketySplitTestSuite(dataset["Behavior"]["HarpLickometerLeft"]), "HarpLickometerLeft" ) From 13617c77902594a5462a81a62464b36c7301e5fb Mon Sep 17 00:00:00 2001 From: Micah Woodard Date: Thu, 13 Aug 2026 11:22:37 -0700 Subject: [PATCH 2/2] fixes name error for environment sensor --- src/Extensions/Hardware.bonsai | 2 +- src/aind_behavior_dynamic_foraging/data_qc/suite.py | 7 ++++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/src/Extensions/Hardware.bonsai b/src/Extensions/Hardware.bonsai index c1730e2..70a5223 100644 --- a/src/Extensions/Hardware.bonsai +++ b/src/Extensions/Hardware.bonsai @@ -196,7 +196,7 @@ RigSchema - HarpSniffDetector + HarpEnvironmentSensor diff --git a/src/aind_behavior_dynamic_foraging/data_qc/suite.py b/src/aind_behavior_dynamic_foraging/data_qc/suite.py index bf65dd9..a0033ae 100644 --- a/src/aind_behavior_dynamic_foraging/data_qc/suite.py +++ b/src/aind_behavior_dynamic_foraging/data_qc/suite.py @@ -72,9 +72,10 @@ def make_qc_runner(dataset: contract.Dataset) -> qc.Runner: _runner.add_suite( qc.harp.HarpLicketySplitTestSuite(dataset["Behavior"]["HarpLickometerRight"]), "HarpLickometerRight" ) - _runner.add_suite( - qc.harp.HarpLicketySplitTestSuite(dataset["Behavior"]["HarpLickometerLeft"]), "HarpLickometerLeft" - ) + if rig.harp_lickometer_left is not None: + _runner.add_suite( + qc.harp.HarpLicketySplitTestSuite(dataset["Behavior"]["HarpLickometerLeft"]), "HarpLickometerLeft" + ) # Add camera qc for camera in dataset["BehaviorVideos"]: