This should happen in [SaveJobReturnValueCallback._save()](https://github.com/Cora4NLP/multi-task-knowledge-transfer/blob/main/src/hydra_callbacks/save_job_return_value.py#L222).