Google colab replication has dependency issues

[mitchellostrow]

I'm trying to run the google colab (link) to replicate the analyses and understand the dataset but I can't get past the imports because it seems like some of the underlying dependencies have changed. Here are a few:

ImportError: cannot import name 'velocity' from 'brainbox.behavior.wheel' (/usr/local/lib/python3.11/dist-packages/brainbox/behavior/wheel.py)

ModuleNotFoundError: No module named 'ibllib.atlas'

Thanks!