Module 'plotly.validators.sankey' has no attribute 'LegendgroupValidator'

I tried the first piece of code in Sankey diagram.

from plotly import graph_objects as go

fig = go.Figure(data=[go.Sankey(
node = dict(
pad = 15,
thickness = 20,
line = dict(color = “black”, width = 0.5),
label = [“A1”, “A2”, “B1”, “B2”, “C1”, “C2”],
color = “blue”
),
link = dict(
source = [0, 1, 0, 2, 3, 3], # indices correspond to labels, eg A1, A2, A2, B1, …
target = [2, 3, 3, 4, 4, 5],
value = [8, 4, 2, 8, 4, 2]
))])

fig.update_layout(title_text=“Basic Sankey Diagram”, font_size=10)
fig.show()

It gives me error:

module ‘plotly.validators.sankey’ has no attribute ‘LegendgroupValidator’

Tried uninstall plotly and the installed it. Doesnt work.

works for me though with plotly==4.1.1