class_docstr: Returns samples with updated `policy_info` (a dictionary).
class_docstr_tok: ["Returns","samples","with","updated","`","policy_info","`","(","a","dictionary",")","."]
code: def _action(
      self, time_step, policy_state, seed
  ):  # pytype: disable=signature-mismatch  # overriding-parameter-count-checks
      action_step = self._wrapped_policy.action(
          time_step, policy_state, seed
      )
      return self._update_info(action_step)
  
file_path: agents/tf_agents/policies/policy_info_updater_wrapper.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from typing import Callable, Dict, Optional, Sequence, Text, Union
  import tensorflow.compat.v2
  from tf_agents.policies import tf_policy
  from tf_agents.trajectories import policy_step
  from tf_agents.typing import types
label_desc: Use image node_id 6 for calling the PolicyInfoUpdaterWrapper obj's underlying member method code with example usage: obj._action(time_step, policy_state, seed) and returns: self