code: def _execute_policy(zip_results_element) -> ps.PolicyStep:
      """Called on each element of zip return value, in _action method."""
      (policy, time_step, policy_state) = zip_results_element
      return policy.action(time_step, policy_state)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from multiprocessing import dummy
  from typing import Sequence, Optional
  import gin
  from tf_agents.policies import py_policy
  from tf_agents.trajectories import policy_step
  from tf_agents.trajectories import time_step
  from tf_agents.typing import types
  from tf_agents.utils import nest_utils
label_desc: Use image node_id 1 for calling a global function with example usage: _execute_policy(zip_results_element) and returns: policy