From 9081643be3e3f50b86de093dd702c75dd9c2b8b6 Mon Sep 17 00:00:00 2001
From: Hammouda Elbez <hammouda.elbez@univ-lille.fr>
Date: Thu, 24 Nov 2022 12:50:19 +0000
Subject: [PATCH] Code improved

---
 src/Modules/General/spark.py | 8 ++++++--
 src/Modules/Synapse/spark.py | 2 +-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/Modules/General/spark.py b/src/Modules/General/spark.py
index 9437936..cb1cf86 100755
--- a/src/Modules/General/spark.py
+++ b/src/Modules/General/spark.py
@@ -78,11 +78,15 @@ class spark(sparkOp):
             try:
                 data = next(col, None)
                 self.g.Accuracy = data["Acc"]
-                self.g.finalLabels = data["NLabel"]
-
             except Exception as e:
                 self.g.Accuracy = "--"
                 print("No accuracy recorded")
+                pass
+            
+            try:
+                self.g.finalLabels = data["NLabel"]
+            except Exception as e:
+                print("No output neuron classes recorded")
                 self.g.finalLabels = None
                 pass
 
diff --git a/src/Modules/Synapse/spark.py b/src/Modules/Synapse/spark.py
index 557b1c9..aef9ded 100755
--- a/src/Modules/Synapse/spark.py
+++ b/src/Modules/Synapse/spark.py
@@ -67,4 +67,4 @@ class spark(sparkOp):
         except Exception as e:
             print("Error:" + str(e))
             traceback.print_exc()
-            pass
+            pass
\ No newline at end of file
-- 
GitLab