code: def normal_with_sigma_1_sampler(mu):
      return np.random.normal(mu, 1)
  
file_path: agents/tf_agents/bandits/environments/stationary_stochastic_per_arm_py_environment_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  import tensorflow
  from tf_agents.bandits.environments import stationary_stochastic_per_arm_py_environment
  from tf_agents.policies import random_py_policy
  from tf_agents.specs import array_spec
label_desc: Use image node_id 1 for calling a global function with example usage: normal_with_sigma_1_sampler(mu) and returns: np