A smart MDA that solves only strongly coupled disciplines and then executes the weakly coupled ones

from __future__ import annotations

from gemseo import configure_logger
from gemseo import create_discipline
from gemseo import create_mda

<RootLogger root (INFO)>

Define a way to display results

def display_result(res, mda_name):
    """Display coupling and output variables in logger.

        res: result of MDA
        mda_name: name of the current MDA
    # names of the coupling variables
    coupling_names = [
    for coupling_var in coupling_names:
        print(f"{mda_name}, coupling variable {coupling_var}: {res[coupling_var]}")

    # names of the output variables
    output_names = ["y_1", "y_2", "y_3", "y_4", "g_1", "g_2", "g_3"]
    for output_name in output_names:
            f"{mda_name}, output variable {output_name}: {res[output_name]}",

Create, execute and post-process MDA

We do not need to specify the inputs, the default inputs of the MDA will be used and computed from the Default inputs of the disciplines

disciplines = create_discipline([
mda = create_mda("MDAChain", disciplines, inner_mda_name="MDAJacobi")
res = mda.execute()
    logscale=[1e-8, 10.0],
    fig_size=(10, 2),
MDAJacobi: residual plot
MDAChain, coupling variable y_11: [0.15591864]
MDAChain, coupling variable y_12: [5.06070636e+04 9.50000000e-01]
MDAChain, coupling variable y_14: [50607.06361395  7306.20262124]
MDAChain, coupling variable y_21: [50607.06361332]
MDAChain, coupling variable y_23: [12194.40917863]
MDAChain, coupling variable y_24: [4.15002177]
MDAChain, coupling variable y_31: [6354.40199355]
MDAChain, coupling variable y_32: [0.50280211]
MDAChain, coupling variable y_34: [1.10754577]
MDAChain, output variable y_1: [5.06070636e+04 7.30620262e+03 9.50000000e-01]
MDAChain, output variable y_2: [5.06070636e+04 1.21944092e+04 4.15002177e+00]
MDAChain, output variable y_3: [1.10754577e+00 6.35440199e+03 5.02802105e-01]
MDAChain, output variable y_4: [535.78213197]
MDAChain, output variable g_1: [ 0.035      -0.00666667 -0.0275     -0.04       -0.04833333 -0.09
 -0.15      ]
MDAChain, output variable g_2: [-0.04]
MDAChain, output variable g_3: [-0.99719789 -0.00280211  0.16206032 -0.02      ]
    INFO - 10:58:30: Requested 10 iterations but the residual history contains only 9, plotting all the residual history.

Total running time of the script: (0 minutes 0.317 seconds)

Gallery generated by Sphinx-Gallery