From 9513380c76c4a92f3281764d3181429b01e92141 Mon Sep 17 00:00:00 2001
From: Martin Chloride <i@martincl2.me>
Date: Thu, 7 May 2020 02:00:31 -0700
Subject: [PATCH] Use tf.abs in compute_spectrogram_tf

To avoid the following error: `Cannot convert a symbolic Tensor (transpose_1:0) to a numpy array`
---
 spleeter/audio/spectrogram.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spleeter/audio/spectrogram.py b/spleeter/audio/spectrogram.py
index 64f4ae8..38d0cf5 100644
--- a/spleeter/audio/spectrogram.py
+++ b/spleeter/audio/spectrogram.py
@@ -44,7 +44,7 @@ def compute_spectrogram_tf(
                 periodic=True,
                 dtype=waveform.dtype) ** window_exponent),
         perm=[1, 2, 0])
-    return np.abs(stft_tensor) ** spec_exponent
+    return tf.abs(stft_tensor) ** spec_exponent
 
 
 def time_stretch(