Error pickling when doing checkpoints

[spartanjoax]

🐛 Describe the bug The checkpoint plugin is giving an error when saving a checkpoint, in the dill library: TypeError: cannot pickle 'ConfigModuleInstance' object

Python version: 3.11.10 Dill version: 0.3.9

🐜 To Reproduce The error can be reproduced by simply running the unit tests on the library:

FAST_TEST=True python -m unittest discover tests -v

🐝 Expected behavior It is expected for the checkpoints to work.