def METHOD_NAME(decorated):
    if inspect.isfunction(decorated):
        return _make_wrapper(feature_name, decorated)
    if inspect.isclass(decorated):
        method_infos = inspect.getmembers(decorated, predicate=_should_wrap)
        _wrap_class_methods(decorated, method_infos, feature_name)
    return decorated