From 4d67121d242ab64cc29f22abafb57d1f9379c9bf Mon Sep 17 00:00:00 2001 From: romi1502 Date: Wed, 1 Jul 2020 15:48:02 +0200 Subject: [PATCH] Get rid of tf.contrib --- spleeter/audio/adapter.py | 2 +- spleeter/audio/spectrogram.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/spleeter/audio/adapter.py b/spleeter/audio/adapter.py index bda1441..720618c 100644 --- a/spleeter/audio/adapter.py +++ b/spleeter/audio/adapter.py @@ -13,7 +13,7 @@ from os.path import exists import numpy as np import tensorflow as tf -from tensorflow.contrib.signal import stft, hann_window +from tensorflow.signal import stft, hann_window # pylint: enable=import-error from .. import SpleeterError diff --git a/spleeter/audio/spectrogram.py b/spleeter/audio/spectrogram.py index 38d0cf5..f16782d 100644 --- a/spleeter/audio/spectrogram.py +++ b/spleeter/audio/spectrogram.py @@ -7,7 +7,7 @@ import numpy as np import tensorflow as tf -from tensorflow.contrib.signal import stft, hann_window +from tensorflow.signal import stft, hann_window # pylint: enable=import-error __email__ = 'research@deezer.com'