ENH Dataset.__setitem__: coerce case-less NDVar to single case

christianbrodbeck · Oct 24, 2021 · c310b1a · c310b1a
1 parent 1f3c10e
commit c310b1a
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/eelbrain/_data_obj.py b/eelbrain/_data_obj.py
@@ -5945,7 +5945,14 @@ def __setitem__(self, index, item):
             # coerce to data-object
             if isdataobject(item) or isinstance(object, Datalist):
                 item.name = index
-                n = 0 if (isinstance(item, NDVar) and not item.has_case) else len(item)
+                if isinstance(item, NDVar) and not item.has_case:
+                    if self.n_cases == 1:  # coerce case-less NDVar
+                        item = item[None]
+                        n = 1
+                    else:
+                        n = 0
+                else:
+                    n = len(item)
             else:
                 if isinstance(item, np.ndarray):
                     if item.ndim == 1: