Function _strategy_registry_handler

examples/simple_trainer.py:274–298

(
    value: Any,
    current: Optional[Strategy],
    path: str,
    apply_fn: Callable[[Any, Dict[str, Any], str], None],
)

}

def _strategy_registry_handler(
    value: Any,
    current: Optional[Strategy],
    path: str,
    apply_fn: Callable[[Any, Dict[str, Any], str], None],
):
    is_training_strategy = False

    if isinstance(value, dict):
        type_name = value.get("type")
        if type_name in _STRATEGY_REGISTRY:
            is_training_strategy = True
        elif isinstance(current, Strategy):
            field_names = {f.name for f in fields(type(current))}
            if set(value.keys()).issubset(field_names):
                is_training_strategy = True
    elif isinstance(value, str) and value in _STRATEGY_REGISTRY:
        is_training_strategy = True
    elif isinstance(current, Strategy):
        is_training_strategy = True

    if not is_training_strategy:
        return current if isinstance(current, Strategy) else value

    return build_from_registry(value, _STRATEGY_REGISTRY, current, apply_fn=apply_fn, path=path)

_REGISTRY_HANDLERS = {
    "strategy": _strategy_registry_handler,