Reformat

DLR-RM · Apr 3, 2023 · e77937f · e77937f
1 parent aa1a64c
commit e77937f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -190,7 +190,7 @@ def dummy_callback(locals_, _globals):
     policy.n_callback_calls = 0  # type: ignore[assignment, attr-defined]
     _, episode_lengths = evaluate_policy(
         policy,  # type: ignore[arg-type]
-        model.get_env(), # type: ignore[arg-type]
+        model.get_env(),  # type: ignore[arg-type]
         n_eval_episodes,
         deterministic=True,
         render=False,
@@ -209,7 +209,7 @@ def dummy_callback(locals_, _globals):
 
     episode_rewards, _ = evaluate_policy(
         policy,  # type: ignore[arg-type]
-        model.get_env(), # type: ignore[arg-type]
+        model.get_env(),  # type: ignore[arg-type]
         n_eval_episodes,
         return_episode_rewards=True,
     )