The transform_extra_arguments was not transmitted through all the wrappers for the checkpoint_preprocessor