This commit is contained in:
Brandon Rising 2023-07-24 18:12:05 -04:00
parent 7073dc0d5d
commit e81601acf3

View File

@ -32,6 +32,7 @@ class Invoker:
datum = batch.data[batch_index] datum = batch.data[batch_index]
for param in datum.keys(): for param in datum.keys():
invocation[param] = datum[param] invocation[param] = datum[param]
# TODO graph.update_node
# Save the execution state # Save the execution state
self.services.graph_execution_manager.set(graph_execution_state) self.services.graph_execution_manager.set(graph_execution_state)
@ -52,10 +53,10 @@ class Invoker:
"""Creates a new execution state for the given graph""" """Creates a new execution state for the given graph"""
new_state = GraphExecutionState(graph=Graph() if graph is None else graph) new_state = GraphExecutionState(graph=Graph() if graph is None else graph)
if graph.batches: if graph.batches:
batch_index = list() batch_indices = list()
for batch in graph.batches: for batch in graph.batches:
batch_index.append(len(batch.data)-1) batch_indices.append(len(batch.data)-1)
new_state.batch_index = batch_index new_state.batch_indices = batch_indices
self.services.graph_execution_manager.set(new_state) self.services.graph_execution_manager.set(new_state)
return new_state return new_state