From 3914b2c5ba8a69e0fd7b69a4a8c682162cb945c4 Mon Sep 17 00:00:00 2001
From: Joel Mathew Thomas <90510078+joelmathewthomas@users.noreply.github.com>
Date: Tue, 28 Jan 2025 20:33:28 +0530
Subject: [PATCH] remove unused parameter

---
 src/separation/convtasnet_wrapper.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/separation/convtasnet_wrapper.py b/src/separation/convtasnet_wrapper.py
index a1d218e..0c08b2a 100644
--- a/src/separation/convtasnet_wrapper.py
+++ b/src/separation/convtasnet_wrapper.py
@@ -1,13 +1,12 @@
 import torch
 from asteroid.models import ConvTasNet
 
-def separate(audio, sr, model_name='mpariente/ConvTasNet_WHAMR_enhsingle'):
+def separate(audio, model_name='mpariente/ConvTasNet_WHAMR_enhsingle'):
     """
     Separates audio into sources using a pretrained Asteroid model.
 
     Args:
         audio (numpy.ndarray): The audio time series (1D numpy array).
-        sr (int): Sampling rate of the audio.
         model_name (str): Name of the pretrained model from Asteroid. Default is 'mpariente/ConvTasNet_WHAMR_enhsingle'.
 
     Returns: