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

fix info not pass issue in PGPolicy (#787)

Browse Source 
close #775
This commit is contained in:
Jiayi Weng 2022-12-24 13:06:54 -08:00 committed by GitHub
parent 4c3791a459
commit 1037627a5b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tianshou/__init__.py
View File

@ -1,6 +1,6 @@
from tianshou import data, env, exploration, policy, trainer, utils from tianshou import data, env, exploration, policy, trainer, utils
__version__ = "0.4.10" __version__ = "0.4.11"
__all__ = [ __all__ = [
"env", "env",

2
tianshou/policy/modelfree/pg.py
View File

@ -105,7 +105,7 @@ class PGPolicy(BasePolicy):
Please refer to :meth:`~tianshou.policy.BasePolicy.forward` for Please refer to :meth:`~tianshou.policy.BasePolicy.forward` for
more detailed explanation. more detailed explanation.
""" """
logits, hidden = self.actor(batch.obs, state=state) logits, hidden = self.actor(batch.obs, state=state, info=batch.info)
if isinstance(logits, tuple): if isinstance(logits, tuple):
dist = self.dist_fn(*logits) dist = self.dist_fn(*logits)
else: else: