Merge pull request #455 from AllenNeuralDynamics:fix-manipulator

Ensure stepper motor is serialized to aind-data-schema

Author: bruno-f-cruz

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(

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"