definitions

transformer_node(func)

Decorator for creating an TransformerTask.

Usage

@transformer_task def my_task(session: Session): ...

Source code in src/lmflux/graphs/mesh/definitions.py

def transformer_node(func:callable):
    """
    Decorator for creating an TransformerTask.

    Usage:
        @transformer_task
        def my_task(session: Session):
            ...
    """
    check_compatible(func, "run", EXPECTED_TRANSFORMER_CALLBACK)
    return TransformerMeshNode(func.__name__, func)