Google Colab demo not working on TPU v5e

Hey sorry if this issue is silly but I could not get the [google demo working](https://colab.research.google.com/github/openxla/xprof/blob/master/docs/xprof_demo.ipynb#scrollTo=UouZW4zcr7sY) Had to `pip install tensorboard `to resolve import error which I am hoping is Ok. ```bash /usr/lib/python3.12/importlib/_bootstrap.py in _find_and_load_unlocked(name, import_) ModuleNotFoundError: No module named 'tensorboard' ``` It clones the xprof demo as expected, but then I get <img width="1691" height="746" alt="Image" src="https://github.com/user-attachments/assets/9302f566-5cb7-4c1d-ba85-5f404a3b25c5" /> I tried different variants of the launch command but no luck ```python %tensorboard --logdir=xprof/demo %tensorboard --logdir=xprof/demo/plugins/profile/tpu-training %tensorboard --logdir="xprof/demo/plugins/profile/tpu-training" ``` Would appreciate any pointers. Thanks !