code: def visit_count_policy(
      root: Node, temperature: float = 1.0
  ) -> types.Probs:
      """Probability weighted by visit^{1/temp} of children nodes."""
      visits = root.children_visits
      if np.sum(visits) == 0:  # uniform policy for zero visits
          visits += 1
      rescaled_visits = visits ** (1 / temperature)
      probs = rescaled_visits / np.sum(rescaled_visits)
      check_numerics(probs)
  
      return probs
  
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 4 for calling a global function with example usage: visit_count_policy(root, temperature) and returns: probs