code: def get_schedule_fn(value_schedule):
      """
      Transform (if needed) learning rate and clip range
      to callable.
  
      :param value_schedule: (callable or float)
      :return: (function)
      """
      # If the passed schedule is a float
      # create a constant function
      if isinstance(value_schedule, (float, int)):
          # Cast to float to avoid errors
          value_schedule = constfn(float(value_schedule))
      else:
          assert callable(value_schedule)
      return value_schedule
  
label_desc: Use image node_id 2 for calling a global function with example usage: get_schedule_fn(value_schedule) and returns: value_schedule