diff --git a/omnisafe/adapter/onpolicy_adapter.py b/omnisafe/adapter/onpolicy_adapter.py index fcef9fe9d..167a87541 100644 --- 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)):