From e8b5e343c7729c7ca52c091045e26101cb525ad4 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Wed, 5 Oct 2022 11:21:30 +0530 Subject: [PATCH] black reformat --- enhancer/inference.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/enhancer/inference.py b/enhancer/inference.py index 838ef5f..1abd8bb 100644 --- a/enhancer/inference.py +++ b/enhancer/inference.py @@ -28,7 +28,7 @@ class Inference: if isinstance(audio, (np.ndarray, torch.Tensor)): assert sr is not None, "Invalid sampling rate!" if len(audio.shape) == 1: - audio = audio.reshape(1,-1) + audio = audio.reshape(1, -1) if isinstance(audio, str): audio = Path(audio) @@ -105,8 +105,7 @@ class Inference: window = get_window(window=window, Nx=data.shape[-1]) window = torch.from_numpy(window).to(data.device) data *= window - step_size = window_size//2 if step_size is None else step_size - + step_size = window_size // 2 if step_size is None else step_size data = data.permute(1, 2, 0) data = F.fold( @@ -134,8 +133,8 @@ class Inference: if isinstance(filename, str): filename = Path(filename) - parent, name = filename.parent, "cleaned_"+filename.name - filename = parent/Path(name) + parent, name = filename.parent, "cleaned_" + filename.name + filename = parent / Path(name) if filename.is_file(): raise FileExistsError(f"file {filename} already exists") else: