From 10eaada080e4326f6ee0d17c09fd707802597a8e Mon Sep 17 00:00:00 2001 From: shahules786 Date: Fri, 23 Sep 2022 18:22:09 +0530 Subject: [PATCH] fix comparison --- enhancer/utils/io.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/enhancer/utils/io.py b/enhancer/utils/io.py index ba3b4d2..80bd1a4 100644 --- a/enhancer/utils/io.py +++ b/enhancer/utils/io.py @@ -59,7 +59,7 @@ class Audio: assert audio.shape[0] == 1, "convert mono only accepts single waveform" audio = audio.reshape(audio.shape[1],audio.shape[2]) - assert audio.shape[0] > audio.shape[1], "expected input format (num_channels,num_samples)" + assert audio.shape[1] >> audio.shape[0], f"expected input format (num_channels,num_samples) got {audio.shape}" num_channels,num_samples = audio.shape if num_channels>1: return audio.mean(axis=0).reshape(1,num_samples)