import torch
from torchmetrics.audio import ShortTimeObjectiveIntelligibility
g = torch.manual_seed(1)
preds = torch.randn(8000)
target = torch.randn(8000)
metric = ShortTimeObjectiveIntelligibility(8000, False)
metric.update(preds, target)
fig_, ax_ = metric.plot()