code: def argmax(values: np.ndarray) -> types.Action:
      """Argmax with random tie-breaking."""
      check_numerics(values)
      max_value = np.max(values)
      return np.int32(
          np.random.choice(np.flatnonzero(values == max_value))
      )
  
file_path: acme/acme/agents/tf/mcts/search.py
imports: import dataclasses
  from typing import Callable, Dict
  from acme.agents.tf.mcts import models
  from acme.agents.tf.mcts import types
  import numpy
label_desc: Use image node_id 5 for calling a global function with example usage: argmax(values) and returns: np