I played around with this but after removing the axis keyword, I think it makes sense to localize the method and the docstring.