From 8078d4ebd3dff43b022a0565e800d4b64aa746fd Mon Sep 17 00:00:00 2001
From: Joel Mathew Thomas <90510078+joelmathewthomas@users.noreply.github.com>
Date: Fri, 27 Dec 2024 19:01:24 +0000
Subject: [PATCH] code cleanup

removed unused parameter from demucs_wrapper
---
 src/separation/demucs_wrapper.py | 2 +-
 tests/test_separation.py         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/separation/demucs_wrapper.py b/src/separation/demucs_wrapper.py
index 0b7fc5b..00958b2 100644
--- a/src/separation/demucs_wrapper.py
+++ b/src/separation/demucs_wrapper.py
@@ -4,7 +4,7 @@ import soundfile as sf
 import tempfile
 from pathlib import Path
 
-def separate_audio_with_demucs(input_file, sample_rate, output_dir: str):
+def separate_audio_with_demucs(input_file, output_dir: str):
     """
     Use subprocess to run Demucs separation on an input audio file.
     Parameters:
diff --git a/tests/test_separation.py b/tests/test_separation.py
index 5241124..c98ddf6 100644
--- a/tests/test_separation.py
+++ b/tests/test_separation.py
@@ -36,7 +36,7 @@ def test_demucs_separation_with_preprocessing():
     new_audio_path = os.path.join(dir_path, f"{file_name}.wav")
     os.rename(temp_audio_path, new_audio_path)
 
-    separate_audio_with_demucs(new_audio_path, samplerate, output_path)
+    separate_audio_with_demucs(new_audio_path, output_path)
 
     # Verify the htdemucs folder exists
     demucs_dir = Path(output_path) / 'htdemucs'