How to make numpy functions accept my object as input