[fix] update requirements

WaveGenAI · Jun 2, 2024 · 234e0c4 · 234e0c4
1 parent 8ecce29
commit 234e0c4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/dataset/process.py b/dataset/process.py
@@ -1,4 +1,5 @@
 import os
+import shutil
 from typing import List
 
 from pydub import AudioSegment
@@ -33,7 +34,7 @@ def __init__(
  self.export_path = export_path
  self.sample_rate = sample_rate
 
- if clean:
+ if clean and self.export_path and os.path.exists(self.export_path):
  self.remove_export_folder()
 
  self.create_export_folder()
@@ -56,7 +57,7 @@ def remove_export_folder(self) -> None:
  """Method to remove the export folder"""
 
  if os.path.exists(self.export_path):
- os.rmdir(self.export_path)
+ shutil.rmtree(self.export_path)
 
  def _split_audio(
  self, audio_path: str, split_windows: int = 32

diff --git a/requirements.txt b/requirements.txt
@@ -11,3 +11,4 @@ evaluate
 librosa
 numpy
 aeneas
+pydub