def METHOD_NAME(edge):
    """
    Function to replace the primitive ops at the edges
    with the DrNAS specific DrNASMixedOp.
    """
    primitives = edge.data.op
    edge.data.set("op", DrNASMixedOp(primitives))