From 06890a43f8d520160b246d55c400940d6808ff2a Mon Sep 17 00:00:00 2001 From: Alistair Johnson Date: Fri, 15 Nov 2019 08:57:10 -0500 Subject: [PATCH 1/2] fix: use joblib rather than sklearn.external --- neurokit/bio/bio_ecg.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/neurokit/bio/bio_ecg.py b/neurokit/bio/bio_ecg.py index 8d0fe3e..494a76d 100644 --- a/neurokit/bio/bio_ecg.py +++ b/neurokit/bio/bio_ecg.py @@ -9,6 +9,7 @@ import mne import biosppy import scipy.signal +import joblib from .bio_ecg_preprocessing import * from .bio_rsp import * @@ -354,9 +355,8 @@ def ecg_signal_quality(cardiac_cycles, sampling_rate, rpeaks=None, quality_model cardiac_cycles = np.array(cardiac_cycles) if quality_model == "default": - model = sklearn.externals.joblib.load(Path.materials() + 'heartbeat_classification.model') - else: - model = sklearn.externals.joblib.load(quality_model) + quality_model = Path.materials() + 'heartbeat_classification.model' + model = joblib.load(quality_model) # Initialize empty dict quality = {} From 7e210859768b3ea507ca693625fad0118f0a37f8 Mon Sep 17 00:00:00 2001 From: Alistair Johnson Date: Fri, 15 Nov 2019 08:57:28 -0500 Subject: [PATCH 2/2] add joblib --- requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index cadb7e3..01dc821 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,4 +11,5 @@ Pillow cvxopt nose coverage -pytest \ No newline at end of file +pytest +joblib