From 45922712d94f2a39ce1e1becf90728e10b688d13 Mon Sep 17 00:00:00 2001 From: Michael Panchenko Date: Fri, 26 Apr 2024 18:14:20 +0200 Subject: [PATCH] Dosctring add return [skip-ci] --- tianshou/data/collector.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tianshou/data/collector.py b/tianshou/data/collector.py index 0b9fd6d..10cf663 100644 --- a/tianshou/data/collector.py +++ b/tianshou/data/collector.py @@ -215,6 +215,7 @@ class BaseCollector(ABC): :param reset_stats: if true, reset the statistics attached to the collector. :param gym_reset_kwargs: extra keyword arguments to pass into the environment's reset function. Defaults to None (extra keyword arguments) + :return: The initial observation and info from the environment. """ obs_NO, info_N = self.reset_env(gym_reset_kwargs=gym_reset_kwargs) if reset_buffer: @@ -731,6 +732,7 @@ class AsyncCollector(Collector): :param reset_stats: if true, reset the statistics attached to the collector. :param gym_reset_kwargs: extra keyword arguments to pass into the environment's reset function. Defaults to None (extra keyword arguments) + :return: The initial observation and info from the environment. """ # This sets the _pre_collect attrs result = super().reset(