TypeError: 'numpy.float64' object cannot be interpreted as an integer

[Vanillaholic]

When running the offered code:

from torch import optim
from pytorch_lightning import Trainer
from asteroid.models import DPRNNTasNet
from asteroid.losses import pairwise_neg_sisdr, PITLossWrapper
from asteroid.data import LibriMix
from asteroid.engine import System
train_loader, val_loader = LibriMix.loaders_from_mini(task="sep_clean", batch_size=16)

model = DPRNNTasNet(n_src=2)

loss = PITLossWrapper(pairwise_neg_sisdr, pit_from="pw_mtx")

optimizer = optim.Adam(model.parameters(), lr=1e-3)

system = System(model, optimizer, loss, train_loader, val_loader)

trainer = Trainer(max_epochs=1, accelerator='mps')
trainer.fit(system)`

it returns an Error:

here are my packages version