Browse Source

[fix] bug

main
silencht 7 months ago
committed by GitHub
parent
commit
fd104241dc
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
  1. 6
      teleop/teleop_hand_and_arm.py

6
teleop/teleop_hand_and_arm.py

@ -66,7 +66,7 @@ if __name__ == '__main__':
parser.add_argument('--headless', action='store_true', help='Enable headless mode (no display)') parser.add_argument('--headless', action='store_true', help='Enable headless mode (no display)')
parser.add_argument('--sim', action = 'store_true', help = 'Enable isaac simulation mode') parser.add_argument('--sim', action = 'store_true', help = 'Enable isaac simulation mode')
parser.add_argument('--record', action = 'store_true', help = 'Enable data recording') parser.add_argument('--record', action = 'store_true', help = 'Enable data recording')
parser.add_argument('--tast-name', type = str, default = 'pick cube', help = 'task name for recording')
parser.add_argument('--task-name', type = str, default = 'pick cube', help = 'task name for recording')
parser.add_argument('--task-goal', type = str, default = 'e.g. pick the red cube on the table.', help = 'task goal for recording') parser.add_argument('--task-goal', type = str, default = 'e.g. pick the red cube on the table.', help = 'task goal for recording')
args = parser.parse_args() args = parser.parse_args()
@ -198,9 +198,9 @@ if __name__ == '__main__':
# record + headless mode # record + headless mode
if args.record and args.headless: if args.record and args.headless:
recorder = EpisodeWriter(task_dir = args.task_dir + args.tast_name, task_goal = args.task_goal, frequency = args.frequency, rerun_log = False)
recorder = EpisodeWriter(task_dir = args.task_dir + args.task_name, task_goal = args.task_goal, frequency = args.frequency, rerun_log = False)
elif args.record and not args.headless: elif args.record and not args.headless:
recorder = EpisodeWriter(task_dir = args.task_dir + args.tast_name, task_goal = args.task_goal, frequency = args.frequency, rerun_log = True)
recorder = EpisodeWriter(task_dir = args.task_dir + args.task_name, task_goal = args.task_goal, frequency = args.frequency, rerun_log = True)
try: try:
logger_mp.info("Please enter the start signal (enter 'r' to start the subsequent program)") logger_mp.info("Please enter the start signal (enter 'r' to start the subsequent program)")

Loading…
Cancel
Save