small_motion

src/behavior_model/data/enums.py

@@ -60,6 +60,7 @@ class Label(Enum):
     HUNCH = 3
     BACK  = 4
     ROLL  = 5
+    SMALL_MOTION = 6
 
 @unique
 class Timeslot(Enum):

src/behavior_model/data/larva_dataset.py

@@ -80,7 +80,8 @@ class LarvaDataset(Dataset):
 
 def import_label(s):
     s = s.decode('utf-8') if isinstance(s, bytes) else s
-    # MaggotUBA only understands BACK, BEND, HUNCH, ROLL, RUN, STOP
+    # MaggotUBA only understands BACK, BEND, HUNCH, ROLL, RUN, STOP.
+    # With the `small_motion` branch, it only understands SMALL_MOTION.
     if s == 'CAST':
         return 'BEND'
     elif s == 'STOP_LARGE':