class_docstr: Returns samples with updated `policy_info` (a dictionary).
class_docstr_tok: ["Returns","samples","with","updated","`","policy_info","`","(","a","dictionary",")","."]
code: def _update_info(self, step):
      if not isinstance(step.info, dict):
          raise ValueError("`step.info` must be a dictionary.")
      current_info = step.info
      current_info.update(self._updater_fn(step))
      return policy_step.PolicyStep(
          step.action, step.state, current_info
      )
  
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 5 for calling the PolicyInfoUpdaterWrapper obj's underlying member method code with example usage: obj._update_info(step) and returns: policy_step