custom_nodes.py
. __init__
and process
methods. You will then be able to add steps with this type in your workflow.yaml
file (the name of your class is the type of the step).model_id
is for instance an argument of the Inference step). To do so, you need to specify those arguments in the signature of the __init__
method. You will also probably want to save those values as attributes to use them in the process
method.process
method should return either:None
if you want to stop the execution of the following steps in the corresponding workflow branch