(sp)
| 260 | sub = p.add_subparsers(dest="cmd", required=True) |
| 261 | |
| 262 | def common(sp): |
| 263 | sp.add_argument("--task", nargs="*", default=["google_robot_pick_coke_can"]) |
| 264 | sp.add_argument("--all", action="store_true") |
| 265 | sp.add_argument("--steps", type=int, default=20) |
| 266 | sp.add_argument("--seed", type=int, default=0) |
| 267 | sp.add_argument("--out-dir", default="/tmp/simpler_trajs") |
| 268 | sp.add_argument("--json", default=None) |
| 269 | |
| 270 | rec = sub.add_parser("record") |
| 271 | common(rec) |