From 16f8e8cac7dbf49203f6774589a7d5779aa346d0 Mon Sep 17 00:00:00 2001
From: Christian Chapman-Bird <c.chapmanbird@btinternet.com>
Date: Mon, 27 Sep 2021 15:10:11 +0100
Subject: [PATCH] heatmap function fix

---
 EMRI_DET/validate.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/EMRI_DET/validate.py b/EMRI_DET/validate.py
index a6476ef..bafcbb2 100644
--- a/EMRI_DET/validate.py
+++ b/EMRI_DET/validate.py
@@ -228,7 +228,7 @@ def grid_heatmap_corner(dataframe, truth_column, pred_column, log=True, ratio=Fa
             ax[j,i].grid(False)
             if i != 0:
                 ax[j,i].yaxis.set_visible(False)
-            if j != 4:
+            if j != nparams-1:
                 ax[j,i].xaxis.set_visible(False)
 
             num += 1
-- 
GitLab