fix(python): Handle generalized ufunc edge cases better

fix(python): Handle generalized ufunc edge cases better #8510