Merge pull request #118 from computational-psychology/fix/whites_targets

fix(whites): enable specifying either 1, or multiple target indices
computational-psychology · Jan 2, 2024 · 8ec7d6e · 8ec7d6e
2 parents 3cba67d + ba552b9
commit 8ec7d6e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/stimupy/stimuli/whites.py b/stimupy/stimuli/whites.py
@@ -118,6 +118,8 @@ def generalized(
     target_bar_mask = mask_targets(
         element_mask=stim["grating_mask"], target_indices=target_indices
     )
+    if isinstance(target_indices, (int, float)):
+        target_indices = (target_indices,)
     stim["target_indices"] = target_indices
 
     # Mask rectangular regions