Skip to content

Commit b78f5ad

Browse files
committed
fixing irregular error in tests when no arrows are drawn
1 parent 5db0215 commit b78f5ad

File tree

3 files changed

+42
-173
lines changed

3 files changed

+42
-173
lines changed

demos/demo_bootstrap.ipynb

Lines changed: 31 additions & 171 deletions
Large diffs are not rendered by default.

src/rsatoolbox/vis/model_plot.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -792,7 +792,11 @@ def plot_arrows(axbar, significant):
792792
k += 1
793793
axbar.plot((i, j), (k, k), 'k-', linewidth=2)
794794
occupied[k-1, i*3+2:j*3+1] = 1
795-
h = occupied.sum(axis=1).nonzero()[0].max()+1
795+
h = occupied.sum(axis=1)
796+
if np.any(h > 0):
797+
h = h.nonzero()[0].max()+1
798+
else:
799+
h = 1
796800
axbar.set_ylim((0, max(expected_n_lines, h)))
797801

798802

tests/test_demo.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -287,7 +287,12 @@ def test_exercise_all(self):
287287
models_flex, rdms_data, train_set, test_set,
288288
ceil_set=ceil_set, method='corr')
289289
# plot results
290-
rsatoolbox.vis.plot_model_comparison(results_3_cv)
290+
rsatoolbox.vis.plot_model_comparison(
291+
results_3_cv,
292+
error_bars=False,
293+
test_pair_comparisons=False,
294+
test_above_0=False,
295+
test_below_noise_ceil=False)
291296

292297
results_3_full = rsatoolbox.inference.bootstrap_crossval(
293298
models_flex, rdms_data, k_pattern=4, k_rdm=2, method='corr', N=5)

0 commit comments

Comments
 (0)