Hey I'm a newbie and experimenting with kedro. I follow the tutorial here - https://kedro.readthedocs.io/en/stable/extend_kedro/custom_datasets.html to create a custom dataset object to store the "sentence-transformer" model for my project. But when I run

kedro viz

to visualize the pipeline, I got this error as the screenshot shows. I configured the "catalog.yml" with the filepath pointing to the dataset class script. Am I doing anything wrong? Or is it a bug in "kedro-viz" that it can't handle custom dataset since it's looking for the specific installed package?