code: def test_vtrace_from_logits():
      _vtrace_from_logits(1)
      _vtrace_from_logits(2)
  
file_path: maze/maze/test/train/impala/test_vtrace.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import gym
  import numpy
  from maze.distributions.distribution_mapper import DistributionMapper
  from maze.perception.perception_utils import convert_to_torch
  from maze.train.trainers.impala import impala_vtrace
label_desc: Use image node_id 9 for calling a global function with example usage: test_vtrace_from_logits() without return types