From bad16ec3c37f4f143ab5acaa2c6f0bc9554b7b17 Mon Sep 17 00:00:00 2001 From: Federico-PizarroBejarano Date: Wed, 4 Dec 2024 12:05:53 -0500 Subject: [PATCH] Reverting to n_episodes=1 --- examples/lqr/lqr_experiment.py | 2 +- examples/mpc/mpc_experiment.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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: