fixed bug when both inputs are not tensors

2023-09-08 08:30:10 +02:00
parent 2643d7999b
commit 1750e551f6
1 changed files with 4 additions and 1 deletions
@@ -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