library installed but not found

Hello, I installed the library using `pip install bertviz`. That completes successfully but import bertviz doesn't work. I wonder if this is because I am doing all of this in a conda environment.