NLP topic modeling by pyLDAvis to visualize results but got an error due to normalized distributions's sum not equal to 1

Question

I am trying to use pyLDAvis to visualize LDA results on databricks.

The env:

Spark NLP version:  2.5.5
Apache Spark version:  2.4.5

I got