@Fahad_hameed Maybe you just forgot to restart your runtime after installing arviz. Remember that whenever you install a new library, you need to restart the runtime as colab will not load installed libaries in the session in which it was installed.
1 Like