Getting TypeError: 'AxesSubplot' object is not subscriptable

[amitli1]

I'm trying to run segmentation model on my wav file (via notebook) and getting error:

from pyannote.audio import Model
from pyannote.audio import Inference
SAMPLE_WAV = "file.wav"
segment_model     = Model.from_pretrained("pyannote/segmentation")
inference         = Inference(segment_model, duration=5.0, step=2.5)
output            = inference(SAMPLE_WAV)
output

Getting error:

pyannote/core/notebook.py:449, in repr_feature(feature)
    447 ylim = (mini - 0.2 * (maxi - mini), maxi + 0.2 * (maxi - mini))
    448 for c, (window, data) in enumerate(feature):
--> 449     ax = axes[c % num_overlap]
    450     step = duration = window.duration / len(data)
    451     frames = SlidingWindow(start=window.start, step=step, duration=duration)

TypeError: 'AxesSubplot' object is not subscriptable

[hbredin]

This is a bug in pyannote.core notebook visualization feature (triggered by the last line that you can remove), not pyannote.audio.

[freshpearYoon]

hi, I am facing the same issue. How can I solve it?