diff --git a/tianshou/data/collector.py b/tianshou/data/collector.py index b57e6f01d..88f6c76ba 100644 --- a/tianshou/data/collector.py +++ b/tianshou/data/collector.py @@ -347,6 +347,7 @@ def collect( info=self.data.info, policy=self.data.policy, env_id=ready_env_ids, + act=self.data.act, ) ) @@ -626,6 +627,7 @@ def collect( truncated=self.data.truncated, info=self.data.info, env_id=ready_env_ids, + act=self.data.act, ) ) except TypeError: @@ -636,6 +638,7 @@ def collect( done=self.data.done, info=self.data.info, env_id=ready_env_ids, + act=self.data.act, ) )