scall12

Commenting in case anyone else runs into this. I had a similar issue on my M1 Mac. Running the code in a different environment (colab) fixed it.