Don't use a multiprocessing.Pool in run_train_models when only specify 1 process