numpy >= 2.0 compatibility