Can you provide your python and numpy versions? It works in google colab if you want to compare from there, you can use '!pip show numpy' to see the versions.