Error pickling when doing checkpoints

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