()
| 124 | |
| 125 | |
| 126 | def main() -> None: |
| 127 | parser = argparse.ArgumentParser(description=__doc__) |
| 128 | parser.add_argument("--tasks", nargs="*", default=["google_robot_pick_coke_can"]) |
| 129 | parser.add_argument("--all", action="store_true") |
| 130 | parser.add_argument("--steps", type=int, default=60) |
| 131 | parser.add_argument("--seed", type=int, default=0) |
| 132 | parser.add_argument("--out", type=str, default="/tmp/simpler_videos") |
| 133 | parser.add_argument("--no-side-by-side", action="store_true", help="render only the passthrough env") |
| 134 | args = parser.parse_args() |
| 135 | |
| 136 | import simpler_env |
| 137 | |
| 138 | tasks = list(simpler_env.ENVIRONMENTS) if args.all else args.tasks |
| 139 | for t in tasks: |
| 140 | try: |
| 141 | p = render_task( |
| 142 | t, steps=args.steps, seed=args.seed, out_dir=args.out, side_by_side=not args.no_side_by_side |
| 143 | ) |
| 144 | print(f"[OK] {t}: {p}") |
| 145 | except Exception as exc: |
| 146 | print(f"[FAIL] {t}: {exc!r}") |
| 147 | |
| 148 | |
| 149 | if __name__ == "__main__": |
no test coverage detected