Selaa lähdekoodia

Merge pull request #130 from facebookresearch/fix_no_redef_error

Fix no-redef mypy error
Ruslan Mavlyutov 1 vuosi sitten
vanhempi
commit
f98c773ced
1 muutettua tiedostoa jossa 2 lisäystä ja 2 poistoa
  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(