In that case you might want to use @googlebot’s suggestion of using callbacks from higher-level wrappers or alternatively derive your custom scheduler from the desired scheduler class and manipulate the step() method as well as its __init__.