wip

PKU-Alignment · May 2, 2024 · a64d3a9 · a64d3a9
1 parent 5bed892
commit a64d3a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/omnisafe/adapter/onpolicy_adapter.py b/omnisafe/adapter/onpolicy_adapter.py
@@ -108,7 +108,7 @@ def rollout(  # pylint: disable=too-many-locals
                 if num_dones:
                     logger.log(
                         f'\nWarning: trajectory cut off when rollout by epoch\
-                            in {num_dones} of {self._env.num_envs} environments.',
+                            in {self._env.num_envs - num_dones} of {self._env.num_envs} environments.',
                     )
 
             for idx, (done, time_out) in enumerate(zip(terminated, truncated)):