From cfbc7813cfa06906c3548d99ba5a879e2ab3db9a Mon Sep 17 00:00:00 2001 From: Borong Zhang Date: Sun, 19 Mar 2023 10:06:18 +0800 Subject: [PATCH] chore: update test --- omnisafe/utils/command_app.py | 3 ++- tests/test_cli.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/omnisafe/utils/command_app.py b/omnisafe/utils/command_app.py index 158c63c58..6c705cfa2 100644 --- 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 index 136779166..dfb27b72d 100644 --- 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