fix(whites): enable specifying either 1, or multiple target indices

Some configuration was causing errors
computational-psychology · Dec 18, 2023 · ba552b9 · ba552b9
1 parent 759f8e8
commit ba552b9
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