🐛 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.