This probably will become obsolete, but in this MR I basically solved some serialization issues.
We no longer need to use functools.partial as callables in the legacy classes.
functools.partial