Просмотр исходного кода

Merge pull request #130 from facebookresearch/fix_no_redef_error

Fix no-redef mypy error
Ruslan Mavlyutov 1 год назад
Родитель
Сommit
f98c773ced
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      scripts/m4t/finetune/dataloader.py

+ 2 - 2
scripts/m4t/finetune/dataloader.py

@@ -162,8 +162,8 @@ class UnitYDataLoader:
             padded_tensors.append(pad_tensor(tensor, padding, "constant", pad_value))
         return torch.stack([tensor for tensor in padded_tensors], dim=0)
 
-    def _prepare_batch(self, samples: List[Dict[str, Any]]) -> MultimodalSeqsBatch:
-        samples = [LangPairSample.from_json(sample) for sample in samples]
+    def _prepare_batch(self, raw_samples: List[Dict[str, Any]]) -> MultimodalSeqsBatch:
+        samples = [LangPairSample.from_json(sample) for sample in raw_samples]
         # input speech
         src_tokens_list = [self._get_source_fbank(sample) for sample in samples]
         src_tokens = self._batch_tensors(