This broke the tutorial and I didn't notice it during code review. This part here makes it overwrite the training YAML path instead of saving the training dataset as a PKL file.