DOC clean pooling_func arg type