From a220f283e3b253f95897ffaf822a2302247b08c9 Mon Sep 17 00:00:00 2001 From: Jim O'Regan Date: Wed, 2 Oct 2024 13:57:12 +0000 Subject: [PATCH] disable consider-using-generator --- matcha/data/text_mel_datamodule.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/matcha/data/text_mel_datamodule.py b/matcha/data/text_mel_datamodule.py index e10dfcb..48f8266 100644 --- a/matcha/data/text_mel_datamodule.py +++ b/matcha/data/text_mel_datamodule.py @@ -234,9 +234,9 @@ class TextMelBatchCollate: def __call__(self, batch): B = len(batch) - y_max_length = max([item["y"].shape[-1] for item in batch]) + y_max_length = max([item["y"].shape[-1] for item in batch]) # pylint: disable=consider-using-generator y_max_length = fix_len_compatibility(y_max_length) - x_max_length = max([item["x"].shape[-1] for item in batch]) + x_max_length = max([item["x"].shape[-1] for item in batch]) # pylint: disable=consider-using-generator n_feats = batch[0]["y"].shape[-2] y = torch.zeros((B, n_feats, y_max_length), dtype=torch.float32)