Apply suggestions from code review
Apply suggestions from code review

Co-authored-by: Eric Wieser <da64e90baec5d084ce3d8e05f3b022ae6d118ff5@gmail.com>
madphysicist/numpy,mattip/numpy,grlee77/numpy,mhvk/numpy,mhvk/numpy,mhvk/numpy,endolith/numpy,mattip/numpy,jakirkham/numpy,pdebuyl/numpy,charris/numpy,charris/numpy,anntzer/numpy,seberg/numpy,simongibbons/numpy,jakirkham/numpy,seberg/numpy,simongibbons/numpy,pdebuyl/numpy,rgommers/numpy,mhvk/numpy,madphysicist/numpy,grlee77/numpy,numpy/numpy,rgommers/numpy,numpy/numpy,endolith/numpy,pdebuyl/numpy,anntzer/numpy,mattip/numpy,grlee77/numpy,grlee77/numpy,simongibbons/numpy,madphysicist/numpy,simongibbons/numpy,simongibbons/numpy,endolith/numpy,seberg/numpy,madphysicist/numpy,rgommers/numpy,pbrod/numpy,madphysicist/numpy,pdebuyl/numpy,rgommers/numpy,jakirkham/numpy,pbrod/numpy,anntzer/numpy,mattip/numpy,endolith/numpy,numpy/numpy,jakirkham/numpy,pbrod/numpy,jakirkham/numpy,pbrod/numpy,anntzer/numpy,charris/numpy,grlee77/numpy,seberg/numpy,mhvk/numpy,charris/numpy,numpy/numpy,pbrod/numpy