Fix bug in removing single effects

cfg/effects/all.yaml

@@ -32,9 +32,9 @@ effects:
   delay:
       _target_: remfx.effects.RandomPedalboardDelay
       sample_rate: ${sample_rate}
-      min_delay_seconds: float = 0.1,
-      max_delay_sconds: float = 1.0,
-      min_feedback: float = 0.05,
-      max_feedback: float = 0.6,
-      min_mix: float = 0.2,
-      max_mix: float = 0.7,
+      min_delay_seconds: 0.1
+      max_delay_sconds: 1.0
+      min_feedback: 0.05
+      max_feedback: 0.6
+      min_mix: 0.2
+      max_mix: 0.7

remfx/datasets.py

@@ -223,10 +223,10 @@ class VocalSet(Dataset):
             effect_indices = torch.arange(len(self.effects_to_remove))
         # Up to max_removed_effects
         if self.max_removed_effects != -1:
-            num_kept_effects = int(torch.rand(1).item() * (self.max_removed_effects))
+            num_removed_effects = int(torch.rand(1).item() * (self.max_removed_effects))
         else:
-            num_kept_effects = len(self.effects_to_remove)
-        effect_indices = effect_indices[: self.max_removed_effects]
+            num_removed_effects = len(self.effects_to_remove)
+        effect_indices = effect_indices[:num_removed_effects]
         # Index in effect settings
         effect_names_to_apply = [self.effects_to_remove[i] for i in effect_indices]
         effects_to_apply = [self.effects[i] for i in effect_names_to_apply]