import torch
from torchmetrics.audio import PerceptualEvaluationSpeechQuality
metric = PerceptualEvaluationSpeechQuality(8000, 'nb')
metric.update(torch.rand(8000), torch.rand(8000))
fig_, ax_ = metric.plot()