Retaining types during numpy operations Disclaimer: I already posted this on Reddit so apologies if it’s just reaching the same community