WIT not rendering fairness indicators in Vertex Workbench (works in Colab)

[rafaelsf80]

I try to run the Fairness_Indicators_Example_Colab.ipynb in Vertex Workbench (user-managed notebook), but when I run this cell, it says Loading widget for ever and nothing is displayed. I restarted the Kernel after the pip installation. TFDV is shown, but not WIT with the fairness indicators. Any suggestions are appreciated.

widget_view.render_fairness_indicator(eval_result=eval_result,
                                      slicing_column=slice_selection,
                                      event_handlers=event_handlers
                                      )
Loading widget...

This problem does NOT happen in Colab.