Reverting to n_episodes=1

Federico-PizarroBejarano · Dec 4, 2024 · bad16ec · bad16ec
1 parent 3085883
commit bad16ec
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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: