hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Dosctring add return [skip-ci]

Browse Source
This commit is contained in:
Michael Panchenko 2024-04-26 18:14:20 +02:00
parent e2e8a699ea
commit 45922712d9
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tianshou/data/collector.py
View File

@ -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(