code: def context_sampling_fn(batch_size, context_dim):
      return np.random.randint(
          -10, 10, [batch_size, context_dim]
      ).astype(np.float32)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import functools
  import gin
  import numpy
  from six.moves import range
  import tensorflow
  from tf_agents.bandits.agents import utils
  from tf_agents.bandits.environments import wheel_py_environment
label_desc: Use image node_id 5 for calling a global function with example usage: context_sampling_fn(batch_size, context_dim) and returns: np