From a48ebdfde8a561161cec0f124099ca0f337ef8b5 Mon Sep 17 00:00:00 2001 From: Tilman Kamp <5991088+tilmankamp@users.noreply.github.com> Date: Tue, 30 Jun 2020 11:34:25 +0200 Subject: [PATCH] Reverb augmentation: Workaround for import problem in scikit-learn dependency of librosa --- training/deepspeech_training/util/augmentations.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/training/deepspeech_training/util/augmentations.py b/training/deepspeech_training/util/augmentations.py index eef58b81..7ac52c41 100644 --- a/training/deepspeech_training/util/augmentations.py +++ b/training/deepspeech_training/util/augmentations.py @@ -341,6 +341,8 @@ class Resample(SampleAugmentation): def apply(self, sample, clock=0.0): # late binding librosa and its dependencies + # pre-importing sklearn fixes https://github.com/scikit-learn/scikit-learn/issues/14485 + import sklearn # pylint: disable=import-outside-toplevel from librosa.core import resample # pylint: disable=import-outside-toplevel sample.change_audio_type(new_audio_type=AUDIO_TYPE_NP) rate = pick_value_from_range(self.rate, clock=clock)