Not sure if applicable here but you could use the `Axis` definition from pandas._typing, if this can accept both integers and "index"/"columns"    Annotated `bool` return value for this and below would be great too