fixed that checkerboard-add_targets converts masks to float

computational-psychology · Dec 12, 2022 · b12cf75 · b12cf75
1 parent 989c42f
commit b12cf75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stimuli/illusions/checkerboards.py b/stimuli/illusions/checkerboards.py
@@ -115,7 +115,7 @@ def add_targets(checkerboard_stim, targets, extend_targets=False, intensity_targ
     img = np.where(mask, intensity_target, checkerboard_stim["img"])
 
     checkerboard_stim["img"] = img
-    checkerboard_stim["mask"] = mask
+    checkerboard_stim["mask"] = mask.astype(int)
     return checkerboard_stim