chore: update test

PKU-Alignment · Mar 19, 2023 · cfbc781 · cfbc781
1 parent 90b114d
commit cfbc781
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/omnisafe/utils/command_app.py b/omnisafe/utils/command_app.py
@@ -183,7 +183,8 @@ def evaluate(
                     )
                     if render:
                         evaluator.render(num_episodes=num_episode)
-                    evaluator.evaluate(num_episodes=num_episode)
+                    else:
+                        evaluator.evaluate(num_episodes=num_episode)
 
 
 @app.command()

diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -76,6 +76,7 @@ def test_eval():
             '1',
             '--height',
             '1',
+            '--no-render',
         ],
     )
     assert result.exit_code == 0