Add action_repeat argument to SB3 example script

examples/sb3_imitation.py

@@ -237,10 +237,8 @@ def close_env():
         print(f"Mean reward after evaluation: {mean_reward}")
 
 except (KeyboardInterrupt, ConnectionError, ConnectionResetError):
-    print(
-        """Training interrupted by user or a ConnectionError. Will save if --save_model_path was
-        used and/or export if --onnx_export_path was used."""
-    )
+    print("""Training interrupted by user or a ConnectionError. Will save if --save_model_path was
+        used and/or export if --onnx_export_path was used.""")
 
 finally:
     handle_onnx_export()

examples/stable_baselines3_example.py

@@ -100,6 +100,12 @@
     default=False,
 )
 parser.add_argument("--speedup", default=1, type=int, help="Whether to speed up the physics in the env")
+parser.add_argument(
+    "--action_repeat",
+    default=None,
+    type=int,
+    help="Similar concept to frame skip, sends action/gets obs every n frames only. Uses sync node setting if not set here.",
+)
 parser.add_argument(
     "--n_parallel",
     default=1,
@@ -158,7 +164,12 @@ def cleanup():
     print("Info: Using --viz without --env_path set has no effect, in-editor training will always render.")
 
 env = StableBaselinesGodotEnv(
-    env_path=args.env_path, show_window=args.viz, seed=args.seed, n_parallel=args.n_parallel, speedup=args.speedup
+    env_path=args.env_path,
+    show_window=args.viz,
+    seed=args.seed,
+    n_parallel=args.n_parallel,
+    speedup=args.speedup,
+    action_repeat=args.action_repeat,
 )
 env = VecMonitor(env)
 
@@ -220,9 +231,7 @@ def func(progress_remaining: float) -> float:
     try:
         model.learn(**learn_arguments)
     except (KeyboardInterrupt, ConnectionError, ConnectionResetError):
-        print(
-            """Training interrupted by user or a ConnectionError. Will save if --save_model_path was
-            used and/or export if --onnx_export_path was used."""
-        )
+        print("""Training interrupted by user or a ConnectionError. Will save if --save_model_path was
+            used and/or export if --onnx_export_path was used.""")
     finally:
         cleanup()