File tree Expand file tree Collapse file tree 1 file changed +0
-32
lines changed
python_workflow_definition/src/python_workflow_definition Expand file tree Collapse file tree 1 file changed +0
-32
lines changed Original file line number Diff line number Diff line change @@ -125,38 +125,6 @@ def write_workflow_json(wg, file_name):
125125 return data
126126
127127
128- def construct_wg_simple (add_x_and_y_func , add_x_and_y_and_z_func ) -> WorkGraph :
129- helper_1 = pickle_node
130- helper_2 = pickle_node
131-
132- add_x_and_y = task .pythonjob (outputs = ["x" , "y" , "z" ])(add_x_and_y_func )
133- add_x_and_y_and_z = task .pythonjob ()(add_x_and_y_and_z_func )
134-
135- # TODO: Create inputs rather than tasks out of data nodes
136- wg = WorkGraph ("wg-simple" )
137-
138- helper_task1 = wg .add_task (helper_1 , name = "x" , value = 1 )
139-
140- helper_task2 = wg .add_task (helper_2 , name = "y" , value = 2 )
141-
142- add_x_and_y_task = wg .add_task (
143- add_x_and_y ,
144- name = "add_x_and_y" ,
145- x = helper_task1 .outputs .result ,
146- y = helper_task2 .outputs .result ,
147- )
148-
149- add_x_and_y_and_z_task = wg .add_task (
150- add_x_and_y_and_z ,
151- name = "add_x_and_y_and_z" ,
152- x = add_x_and_y_task .outputs .x ,
153- y = add_x_and_y_task .outputs .y ,
154- z = add_x_and_y_task .outputs .z ,
155- )
156-
157- return wg
158-
159-
160128def construct_wg_qe (
161129 get_bulk_structure ,
162130 calculate_qe ,
You can’t perform that action at this time.
0 commit comments