Fix runs bug

DAS/visualizer.py

@@ -6,12 +6,12 @@ import matplotlib.pyplot as plt
 import numpy as np
 import seaborn as sns
 from itertools import combinations
-from config_example import runs
 
 class Visualizer:
 
-    def __init__(self, execID):
+    def __init__(self, execID, config):
         self.execID = execID
+        self.config = config
         self.folderPath = "results/"+self.execID
         self.parameters = ['run', 'blockSize', 'failureRate', 'numberNodes', 'netDegree', 'chi', 'vpn1', 'vpn2', 'class1ratio', 'bwUplinkProd', 'bwUplink1', 'bwUplink2']
         self.minimumDataPoints = 2
@@ -146,8 +146,8 @@ class Visualizer:
         """Plot and store the 2D heatmaps in subfolders"""
         data= self.plottingData()
         """Average the runs if needed"""
-        if(len(runs) > 1):
-            data = self.averageRuns(data, len(runs))
+        if(len(self.config.runs) > 1):
+            data = self.averageRuns(data, len(self.config.runs))
         filteredKeys = self.similarKeys(data)
         vmin, vmax = 0, self.maxTTA
         print("Plotting heatmaps...")

study.py

@@ -60,7 +60,7 @@ def study():
         sim.logger.info("Results dumped into results/%s/" % (execID), extra=sim.format)
 
     if config.visualization:
-        vis = Visualizer(execID)
+        vis = Visualizer(execID, config)
         vis.plotHeatmaps()