diff --git a/flowrep/workflow.py b/flowrep/workflow.py index 8d38c3a..bbbbb11 100644 --- a/flowrep/workflow.py +++ b/flowrep/workflow.py @@ -1302,7 +1302,7 @@ def get_hashed_node_dict( if "function" not in nodes_dict[node]: raise ValueError("Hashing works only on flat data") data_dict = { - "nodes": _get_function_metadata(nodes_dict[node]["function"]), + "node": _get_function_metadata(nodes_dict[node]["function"]), "inputs": {}, "outputs": list(nodes_dict[node]["outputs"].keys()), } @@ -1324,7 +1324,7 @@ def get_hashed_node_dict( else: value = predecessor data_dict["inputs"][key] = value - data_dict["nodes"]["connected_inputs"] = connected_inputs + data_dict["node"]["connected_inputs"] = connected_inputs return data_dict diff --git a/tests/unit/test_workflow.py b/tests/unit/test_workflow.py index 2bea13c..e3cba4c 100644 --- a/tests/unit/test_workflow.py +++ b/tests/unit/test_workflow.py @@ -928,7 +928,7 @@ def workflow_with_data(a=10, b=20): self.assertEqual( data_dict, { - "nodes": { + "node": { "module": add.__module__, "qualname": "add", "version": "not_defined", @@ -945,7 +945,7 @@ def workflow_with_data(a=10, b=20): self.assertEqual( data_dict, { - "nodes": { + "node": { "module": multiply.__module__, "qualname": "multiply", "version": "not_defined",