consider node types during generation

2 files changed, 4 insertions, 2 deletions

diff --git a/Metrics/Metrics-Calculation/metrics_plot/utils/GraphType.py b/Metrics/Metrics-Calculation/metrics_plot/utils/GraphType.py
index 827c2a5e..d6bdad0d 100644
--- a/Metrics/Metrics-Calculation/metrics_plot/utils/GraphType.py
+++ b/Metrics/Metrics-Calculation/metrics_plot/utils/GraphType.py
@@ -11,8 +11,9 @@ class GraphCollection:
         self.nas = []
         self.mpcs = []
         self.name = name
-        self.size = number
         models = reader.readmultiplefiles(path, number, shouldShuffle)
+        print(len(models))
+        self.size = len(models)
         for i in range(len(models)):
             contents, out_d, na, mpc = reader.getmetrics(models[i])
             self.out_ds.append(out_d)
diff --git a/Metrics/Metrics-Calculation/metrics_plot/utils/readCSV.py b/Metrics/Metrics-Calculation/metrics_plot/utils/readCSV.py
index b27a0ffc..33378ce5 100644
--- a/Metrics/Metrics-Calculation/metrics_plot/utils/readCSV.py
+++ b/Metrics/Metrics-Calculation/metrics_plot/utils/readCSV.py
@@ -29,7 +29,8 @@ def readcsvfile(filename):
                 try:
                     contents[arr[0]] = list(map(int, arr[1:]))
                 except:
-                    print(arr[0], ' is not integer')
+                    message = arr[0], ' is not integer'
+                    #print(message)
     f.close()
     return contents