FIX morph_source_space(): VolumeSourceSpace

christianbrodbeck · May 18, 2021 · 01fd78b · 01fd78b
1 parent b6e9342
commit 01fd78b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/eelbrain/_mne.py b/eelbrain/_mne.py
@@ -19,7 +19,7 @@
 except ImportError:
     from mne import compute_morph_matrix
 
-from ._data_obj import NDVar, Space, SourceSpace, VolumeSourceSpace
+from ._data_obj import NDVar, Space, SourceSpaceBase, SourceSpace, VolumeSourceSpace
 from ._utils.numpy_utils import index
 
 
@@ -474,7 +474,7 @@ def morph_source_space(
         # convert to boolean mask (morphing involves interpolation, so the output is in floats)
         mask = round(mask).astype(bool)
     """
-    if isinstance(data, SourceSpace):
+    if isinstance(data, SourceSpaceBase):
         source, ndvar, axis = data, None, None
     else:
         ndvar = data