FIX SequencePlotter: process label subject

eelbrain/plot/_brain.py

@@ -1193,7 +1193,12 @@ def _assert_has_frame_dim(self):
         elif self._frame_dim is False:
             raise RuntimeError("Only applies to SequencePlotters with 2d NDVars")
 
-    def set_brain(self, source):
+    def set_brain(
+            self,
+            source: SourceSpace = None,
+            subject: str = None,
+            subjects_dir: PathArg = None,
+    ):
         """Set the brain model on which to plot
 
         This is usually handled automatically, and only needs to be invoked
@@ -1202,18 +1207,27 @@ def set_brain(self, source):
 
         Parameters
         ----------
-        source : SourceSpace
+        source
             Brain model on which to plot.
+        subject
+            Specify subject directly.
+        subjects_dir
+            Specify the subjects directory directly.
         """
-        if not isinstance(source, SourceSpace):
+        if isinstance(source, SourceSpace):
+            subject = source.subject
+            subjects_dir = source.subjects_dir
+        elif source is not None:
             raise TypeError(f"{source!r}")
-        elif self._subject is None:
-            self._subject = source.subject
-            self._subjects_dir = source.subjects_dir
-        elif source.subject != self._subject:
-            raise ValueError(f"NDVar has different subject ({source.subject}) than previously added data ({self._subject})")
-        elif source.subjects_dir != self._subjects_dir:
-            raise ValueError(f"NDVar has different subjects_dir ({source.subjects_dir}) than previously added data ({self._subjects_dir})")
+
+        if self._subject is None:
+            self._subject = subject
+        elif subject != self._subject:
+            raise ValueError(f"Different subject ({subject}) than previously added ({self._subject})")
+        if self._subjects_dir is None:
+            self._subjects_dir = subjects_dir
+        elif subjects_dir != self._subjects_dir:
+            raise ValueError(f"different subjects_dir ({subjects_dir}) than previously added ({self._subjects_dir})")
 
     def _n_items(self):
         if self._bin_kind == SPLayer.SEQUENCE:
@@ -1268,6 +1282,7 @@ def add_label(
             **kwargs,
     ):
         """"""
+        self.set_brain(subject=mne_label.subject)
         if overlay:
             kind = SPLayer.OVERLAY
         else: