diff --git a/examples/lqr/lqr_experiment.py b/examples/lqr/lqr_experiment.py index faff7934b..e21807185 100644 --- a/examples/lqr/lqr_experiment.py +++ b/examples/lqr/lqr_experiment.py @@ -15,7 +15,7 @@ from safe_control_gym.utils.registration import make -def run(gui=False, plot=True, n_episodes=2, n_steps=None, save_data=False): +def run(gui=False, plot=True, n_episodes=1, n_steps=None, save_data=False): '''The main function running LQR and iLQR experiments. Args: diff --git a/examples/mpc/mpc_experiment.py b/examples/mpc/mpc_experiment.py index 139ad8032..5d69738a3 100644 --- a/examples/mpc/mpc_experiment.py +++ b/examples/mpc/mpc_experiment.py @@ -14,7 +14,7 @@ from safe_control_gym.utils.registration import make -def run(gui=False, plot=True, n_episodes=2, n_steps=None, save_data=False): +def run(gui=False, plot=True, n_episodes=1, n_steps=None, save_data=False): '''The main function running MPC and Linear MPC experiments. Args: