diff --git a/tianshou/data/collector.py b/tianshou/data/collector.py index b57e6f0..88f6c76 100644 --- a/tianshou/data/collector.py +++ b/tianshou/data/collector.py @@ -347,6 +347,7 @@ class Collector(object): info=self.data.info, policy=self.data.policy, env_id=ready_env_ids, + act=self.data.act, ) ) @@ -626,6 +627,7 @@ class AsyncCollector(Collector): truncated=self.data.truncated, info=self.data.info, env_id=ready_env_ids, + act=self.data.act, ) ) except TypeError: @@ -636,6 +638,7 @@ class AsyncCollector(Collector): done=self.data.done, info=self.data.info, env_id=ready_env_ids, + act=self.data.act, ) )