From e1ee84965b93ec820bac9c64085aa272df828569 Mon Sep 17 00:00:00 2001
From: Nikolai <osterei33@gmx.de>
Date: Wed, 22 Aug 2018 09:15:15 +0200
Subject: [PATCH] fixing plot_NN_2D.py for projects without mask values

---
 scripts/plot_NN_2D.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/plot_NN_2D.py b/scripts/plot_NN_2D.py
index 1555095..f5f9fee 100755
--- a/scripts/plot_NN_2D.py
+++ b/scripts/plot_NN_2D.py
@@ -82,7 +82,7 @@ total_weights = c.w_test*np.array(c.class_weight)[c.y_test.astype(int)]
 
 try:
     mask_value = c.mask_value
-except NameError:
+except AttributeError:
     mask_value = None
 
 # varx_test = c.x_test[:,varx_index]
-- 
GitLab