Fix wrong signature for main in examples/examples_movement_primitives.py

ALRhub · Nov 26, 2023 · 374d72d · 374d72d
1 parent d6f42cf
commit 374d72d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/fancy_gym/examples/examples_movement_primitives.py b/fancy_gym/examples/examples_movement_primitives.py
@@ -250,8 +250,7 @@ def example_fully_custom_mp_alternative(seed=1, iterations=1, render=True):
         pass
 
 
-def main():
-    render = False
+def main(render=False):
     # DMP
     example_mp("fancy_DMP/HoleReacher-v0", seed=10, iterations=5, render=render)