distributed_flow.py

from prefect import Flow

with Flow("distributed") as flow:
    X_train, X_test, y_train, y_test = create_data()
    models = get_models()
    training_runs = train_model.map(models, unmapped(X_train), 
                                    unmapped(X_test), unmapped(y_train), 
                                    unmapped(y_test))
    get_results(training_runs)