From fceaa4763df1eaaf13fbbe542b46bac2032abe93 Mon Sep 17 00:00:00 2001 From: Joris Vincent Date: Wed, 15 Jun 2022 11:01:36 +0200 Subject: [PATCH] plot_stims takes argument for showing masks --- stimuli/utils/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/stimuli/utils/utils.py b/stimuli/utils/utils.py index 6d4b9ce0..3cb48823 100644 --- a/stimuli/utils/utils.py +++ b/stimuli/utils/utils.py @@ -486,7 +486,7 @@ def plot_stim(stim, mask=False, stim_name="stim", ax=None): return ax -def plot_stimuli(stims): +def plot_stimuli(stims, mask=False): import math # Plot each stimulus+mask @@ -494,7 +494,7 @@ def plot_stimuli(stims): F = plt.figure(figsize=(n_stim * 3, n_stim * 3)) for i, (stim_name, stim) in enumerate(stims.items()): ax = F.add_subplot(n_stim, n_stim, i + 1) - plot_stim(stim, True, stim_name=stim_name, ax=ax) + plot_stim(stim, mask, stim_name=stim_name, ax=ax) plt.tight_layout() plt.show()