diff --git a/coffeine/power_features.py b/coffeine/power_features.py index 78ac510..6d94c28 100644 --- a/coffeine/power_features.py +++ b/coffeine/power_features.py @@ -2,7 +2,12 @@ import numpy as np import pandas as pd from scipy.stats import trim_mean -from pyriemann.estimation import CospCovariances +try: + from pyriemann.estimation import CospCovariances +except ImportError: + # CospCovariances was renamed to CoSpectra in pyriemann 0.6 + # and removed in pyriemann 0.7 (PR #327) + from pyriemann.estimation import CoSpectra as CospCovariances import mne from mne.io import BaseRaw from mne.epochs import BaseEpochs