So that schedule_func() calls get_piecewise_constant_schedule() with correctly named kwarg.
Attach a file by drag & drop or click to upload
