Be careful on the inputs of the graph_builder task.

[superstar54]

In principle, a graph_builder task can accept non orm.Data as input. For example, when users use PythonJob, we allow them to pass raw Python data as input, so that they do not need to know orm.Data. However, when the graph_builder task use the output of another task as its input, the input value could be a orm.Data, therefore, the graph_builder task handles both cases.

@task.graph_builder()
def relax_workgraph(
    atoms: Atoms = None,
    command: str = "pw.x",
):

In the above example, the input atoms could be an ase.Atoms or aiida_workgraph.orm.atoms.AtomsData.