Add 'do_train' check to cpt

[BramVanroy]

I forgot the do_train when creating the cpt script, or for some reason left it out. But I think it would be ueseful to add it still.

[HuggingFaceDocBuilderDev]

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.