add wine task to constants

agilexrobotics · Nov 2, 2023 · ce40fff · ce40fff
1 parent 9a325e6
commit ce40fff
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/aloha_scripts/constants.py b/aloha_scripts/constants.py
@@ -38,6 +38,12 @@
         'num_episodes': None,
         'episode_len': 1500,
         'camera_names': ['cam_high', 'cam_left_wrist', 'cam_right_wrist']
+    },
+    'aloha_mobile_wipe_wine':{
+        'dataset_dir': DATA_DIR + '/aloha_mobile_wipe_wine',
+        'num_episodes': None,
+        'episode_len': 1300,
+        'camera_names': ['cam_high', 'cam_left_wrist', 'cam_right_wrist']
     }
 }