pub async fn execute( command: TrainCommands, _config: &Config, _output_format: &str, ) -> Result<()>