From 1750e551f6363016e5d802e91e3c4a22317a927b Mon Sep 17 00:00:00 2001 From: Jaikinator Date: Fri, 8 Sep 2023 08:30:10 +0200 Subject: [PATCH] fixed bug when both inputs are not tensors --- autotranscript/audio.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/autotranscript/audio.py b/autotranscript/audio.py index 04feb1d..4d457b6 100644 --- a/autotranscript/audio.py +++ b/autotranscript/audio.py @@ -98,7 +98,10 @@ class AudioProcessor: """ start = int(start * self.sr) - end = int(torch.ceil(end * self.sr)) + if (isinstance(end, float) or isinstance(end, int)) and isinstance(self.sr, int): + end = int(np.ceil(end * self.sr)) + else: + end = int(torch.ceil(end * self.sr)) return self.waveform[start:end] @staticmethod