Import fails by default in colab

[jacobcd52]

Describe the bug
I run this code in an empty, default colab notebook:

!pip install transformer_lens
import transformer_lens

I get this error:

ValueError: numpy.dtype size changed, may indicate binary incompatibility. Expected 96 from C header, got 88 from PyObject

Code example
See above

System Info
I opened up google colab with entirely default settings, created a single code cell and ran the above code.

Additional context
None

Checklist
[x] I have checked that there is no similar issue in the repo (required)

[jakobhansen-blai]

I believe this is due to TL installing Numpy v1 instead of the v2 which is installed by default in Colab. I've been able to work around it by restarting the session and running everything from the top. It would be very nice if Transformer Lens supported Numpy v2, though.

[s-adhit]

@jakobhansen-blai
After checking the repo, no significant changes are needed for V2. You could probably try using ruff linter to check and migrate it to V2.