def METHOD_NAME(alg_cls: Type[Algorithm]) -> Callable:
    if alg_cls is BlurPool:
        return apply_blurpool
    if alg_cls is Factorize:
        return apply_factorization
    if alg_cls is GhostBatchNorm:
        return apply_ghost_batchnorm
    if alg_cls is SqueezeExcite:
        return apply_squeeze_excite
    if alg_cls is StochasticDepth:
        return apply_stochastic_depth
    if alg_cls is ChannelsLast:
        return apply_channels_last
    raise ValueError(f'Unknown algorithm class {alg_cls}')