Bug resolve

changed file generic.py and style.py
replaced if axis in [...] with get_axis_number(axis)

Hello @soni-muskan1234! Thanks for updating this PR. We checked the lines you've touched for [PEP 8](https://www.python.org/dev/peps/pep-0008) issues, and found:



















There are currently no PEP 8 issues detected in this Pull Request. Cheers! :beers: 

##### Comment last updated at 2021-08-25 13:30:59 UTC Please run pre-commit checks locally. This won't pass the ci @soni-muskan1234 let us know if you want to carry on working on this. See comments from maintainers above on next steps. thanks @soni-muskan1234 - closing as the issue is now resolved