fix info not pass issue in PGPolicy (#787)

close #775
thu-ml · Dec 24, 2022 · 1037627 · 1037627
1 parent 4c3791a
commit 1037627
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tianshou/__init__.py b/tianshou/__init__.py
@@ -1,6 +1,6 @@
 from tianshou import data, env, exploration, policy, trainer, utils
 
-__version__ = "0.4.10"
+__version__ = "0.4.11"
 
 __all__ = [
     "env",

diff --git a/tianshou/policy/modelfree/pg.py b/tianshou/policy/modelfree/pg.py
@@ -105,7 +105,7 @@ def forward(
             Please refer to :meth:`~tianshou.policy.BasePolicy.forward` for
             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):
             dist = self.dist_fn(*logits)
         else: