FIX plot.brain: allow ndvar with dim other than time

christianbrodbeck · Dec 3, 2021 · ed6d864 · ed6d864
1 parent 40cb7d4
commit ed6d864
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/eelbrain/plot/_brain_object.py b/eelbrain/plot/_brain_object.py
@@ -15,7 +15,7 @@
 import numpy as np
 import scipy.ndimage
 
-from .._data_obj import NDVar, SourceSpace, asndvar
+from .._data_obj import NDVar, SourceSpace, UTS, asndvar
 from .._exceptions import KeysMissing
 from .._text import ms
 from .._types import PathArg
@@ -403,7 +403,7 @@ def add_ndvar(self, ndvar, cmap=None, vmin=None, vmax=None,
             self.remove_data()
 
         # make sure time axis is compatible with existing data
-        if time_dim is not None:
+        if isinstance(time_dim, UTS):
             self._init_time_dim(time_dim)
 
         # find colormap parameters