diff --git a/src/aind_behavior_vr_foraging/data_mappers/_rig.py b/src/aind_behavior_vr_foraging/data_mappers/_rig.py index 7572cb2c..c979c2d7 100644 --- a/src/aind_behavior_vr_foraging/data_mappers/_rig.py +++ b/src/aind_behavior_vr_foraging/data_mappers/_rig.py @@ -375,6 +375,15 @@ def _get_all_components_and_connections( is_clock_generator=False, ) + harp_stepper_motor = devices.HarpDevice( + name=validate_name(rig, "manipulator"), + harp_device_type=devices.HarpDeviceType.STEPPERDRIVER, + manufacturer=devices.Organization.OEPS, + is_clock_generator=False, + ) + + _components.append(harp_stepper_motor) + if rig.harp_sniff_detector is not None: _components.append( devices.HarpDevice( @@ -492,7 +501,7 @@ def _get_lickspout_assembly( model="328-300-00", travel=Decimal("30"), travel_unit=units.SizeUnit.CM, - notes="This stage is driven by the HarpStepperDriver device.", + notes="This stage is driven by the manipulator device.", ) return devices.LickSpoutAssembly( diff --git a/src/aind_behavior_vr_foraging/data_mappers/_utils.py b/src/aind_behavior_vr_foraging/data_mappers/_utils.py index b6b7c4c5..60a713aa 100644 --- a/src/aind_behavior_vr_foraging/data_mappers/_utils.py +++ b/src/aind_behavior_vr_foraging/data_mappers/_utils.py @@ -106,7 +106,7 @@ class TrackedDevices(enum.StrEnum): TORQUE_SENSOR = "torque_sensor" ROTARY_ENCODER = "rotary_encoder" ENCLOSURE = "behavior_enclosure" - MOTORIZED_STAGE = "manipulator" + MOTORIZED_STAGE = "motorized_stage" LICK_SPOUT = "lick_spout" SCREEN = "screen" COMPUTER = "computer"