FIX NDVar.__setitem__: boolean index

christianbrodbeck · Jun 12, 2021 · 748a3ed · 748a3ed
1 parent 928eb3c
commit 748a3ed
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/eelbrain/_data_obj.py b/eelbrain/_data_obj.py
@@ -3511,7 +3511,10 @@ def __setitem__(self, key, value):
         index = self._array_index(key)
         if isinstance(value, NDVar):
             value = self._ialign(value, index)
-        self.x[index] = value
+        if isinstance(index, np.ndarray) and index.dtype.kind == 'i':
+            self.x[index] = value.ravel()
+        else:
+            self.x[index] = value
 
     def __len__(self):
         return len(self.x)