From 40c7636468b47bfe842f819d6e703f760c367423 Mon Sep 17 00:00:00 2001
From: David Robert Verelst <dave@dtu.dk>
Date: Fri, 31 Mar 2017 16:12:33 +0200
Subject: [PATCH] prepost.mplutils: wafo as optional dependency for peak
detection
---
wetb/prepost/mplutils.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/wetb/prepost/mplutils.py b/wetb/prepost/mplutils.py
index 02e60e24..8e8693c1 100644
--- a/wetb/prepost/mplutils.py
+++ b/wetb/prepost/mplutils.py
@@ -24,7 +24,11 @@ import numpy as np
import matplotlib as mpl
# use a headless backend
from matplotlib.backends.backend_agg import FigureCanvasAgg as FigCanvas
-#import wafo
+# wafo is an optional dependency only required for non default PSD peak marking
+try:
+ import wafo
+except ImportError:
+ pass
def make_fig(nrows=1, ncols=1, figsize=(12,8), dpi=120):
@@ -267,7 +271,7 @@ def match_yticks(ax1, ax2, nr_ticks_forced=None, extend=False):
def time_psd(results, labels, axes, alphas=[1.0, 0.7], colors=['k-', 'r-'],
NFFT=None, res_param=250, f0=0, f1=None, nr_peaks=10, min_h=15,
- mark_peaks=True):
+ mark_peaks=False):
"""
Plot time series and the corresponding PSD of the channel.
--
GitLab