code: def piecewise_constant(chosen_action, context, reward):
      return reward[int(chosen_action) - 1]
  
label_desc: Use image node_id 2 for calling a global function with example usage: piecewise_constant(chosen_action, context, reward) and returns: reward