From 00317bc4f369b59f1873ca835e11d4762b52c938 Mon Sep 17 00:00:00 2001 From: superstar54 Date: Wed, 4 Dec 2024 10:29:06 +0100 Subject: [PATCH] update docs --- docs/gallery/autogen/quick_start.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/docs/gallery/autogen/quick_start.py b/docs/gallery/autogen/quick_start.py index ad41ad1f..d7f0b7a8 100644 --- a/docs/gallery/autogen/quick_start.py +++ b/docs/gallery/autogen/quick_start.py @@ -357,6 +357,22 @@ def multiply(x, y): generate_node_graph(wg.pk) +###################################################################### +# One can also set task inputs from an AiiDA process builder directly. +# + +from aiida.calculations.arithmetic.add import ArithmeticAddCalculation + +builder = ArithmeticAddCalculation.get_builder() +builder.code = code +builder.x = Int(2) +builder.y = Int(3) + +wg = WorkGraph("test_set_inputs_from_builder") +add1 = wg.add_task(ArithmeticAddCalculation, name="add1") +add1.set_from_builder(builder) + + ###################################################################### # Graph builder # -------------