add env_id in preprocess fn #391
Conversation
54916a5 to
92e8441
Compare
Codecov Report
@@ Coverage Diff @@
## master #391 +/- ##
=======================================
Coverage 94.65% 94.65%
=======================================
Files 57 57
Lines 3758 3758
=======================================
Hits 3557 3557
Misses 201 201
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
externalsupplierstaff
left a comment
There was a problem hiding this comment.
Also change
The "preprocess_fn" is a function called before the data has been added to the
buffer with batch format. It will receive with only "obs" when the collector resets
the environment, and will receive five keys "obs_next", "rew", "done", "info", and
"policy" in a normal env step. It returns either a dict or a
:class:~tianshou.data.Batchwith the modified keys and values. Examples are in
"test/base/test_collector.py".
By
The "preprocess_fn" is a function called before the data has been added to the
buffer with batch format. It will receive only "obs" and "env_id" when the collector resets
the environment, and will receive six keys "obs_next", "rew", "done", "info", "policy" and "env_id" in a normal env step. It returns either a dict or a
:class:~tianshou.data.Batchwith the modified keys and values. Examples are in
"test/base/test_collector.py".
close #390