Update musdb_moisesdb_datamodule.py

JusperLee · Oct 30, 2024 · b70911d · b70911d
1 parent fc75463
commit b70911d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/look2hear/datas/musdb_moisesdb_datamodule.py b/look2hear/datas/musdb_moisesdb_datamodule.py
@@ -120,7 +120,7 @@ def __getitem__(self, idx: int) -> Tuple[torch.Tensor, torch.Tensor]:
             start = random.randint(0, music_wav.shape[-1] - self.segments)
             ori_wav = music_wav[:, start:start+self.segments]
 
-        codec_wav = codec_simu(ori_wav, sr=self.sr, codec_type=self.codec_type, options=self.codec_options)
+        codec_wav = codec_simu(ori_wav, sr=self.sr, options=self.codec_options)
 
         max_scale = max(ori_wav.abs().max(), codec_wav.abs().max())
 
@@ -212,4 +212,4 @@ def val_dataloader(self) -> DataLoader:
             num_workers=self.hparams.num_workers,
             shuffle=False,
             pin_memory=True,
-        )
+        )