formatting and change util script to be able to load directories as symlinks

mu_map/random_search/eval/util.py

@@ -12,11 +12,12 @@ import pandas as pd
 from mu_map.random_search.cgan import load_params
 
 
-SIZE_DEFAULT = 12
+SIZE_DEFAULT = 16
 plt.rc("font", family="Roboto")  # controls default font
 plt.rc("font", weight="normal")  # controls default font
 plt.rc("font", size=SIZE_DEFAULT)  # controls default text sizes
-plt.rc("axes", titlesize=18)  # fontsize of the axes title
+# plt.rc("axes", titlesize=18)  # fontsize of the axes title
+plt.rc("axes", titlesize=22)  # fontsize of the axes title
 
 
 class ColorList:
@@ -112,9 +113,9 @@ def load_data(
         a dict mapping the iteration number to "measures", "params", and "dir"
     """
     dirs_run = sorted(os.listdir(dir_random_search))
+    dirs_run = filter(lambda f: f.isdigit(), dirs_run)
     dirs_run = map(lambda f: os.path.join(dir_random_search, f), dirs_run)
     dirs_run = filter(lambda f: os.path.isdir(f), dirs_run)
-    dirs_run = filter(lambda f: not os.path.islink(f), dirs_run)
     dirs_run = map(lambda f: os.path.basename(f), dirs_run)
 
     data = {}
@@ -239,7 +240,7 @@ class TablePrinter:
                 for header, column in table.items()
             ]
         )
-        lenghtes = dict(
+        lengthes = dict(
             [
                 (header, max(len(header), *map(len, column)))
                 for header, column in table.items()
@@ -247,19 +248,19 @@ class TablePrinter:
         )
 
         line_top = f"{self.vert}{self.t_down}{self.vert}".join(
-            map(lambda header: self.vert * lenghtes[header], headers)
+            map(lambda header: self.vert * lengthes[header], headers)
         )
         line_top = self.top_left + self.vert + line_top + self.vert + self.top_right
         print(line_top)
 
         line_headers = f" {self.hori} ".join(
-            map(lambda header: f"{header:>{lenghtes[header]}}", table.keys())
+            map(lambda header: f"{header:>{lengthes[header]}}", table.keys())
         )
         line_headers = self.hori + " " + line_headers + " " + self.hori
         print(line_headers)
 
         line_mid = f"{self.vert}{self.cross}{self.vert}".join(
-            map(lambda header: self.vert * lenghtes[header], headers)
+            map(lambda header: self.vert * lengthes[header], headers)
         )
         line_mid = self.t_left + self.vert + line_mid + self.vert + self.t_right
         print(line_mid)
@@ -267,7 +268,7 @@ class TablePrinter:
         for i in range(len(table[headers[0]])):
             values = map(
                 lambda header: self.color(
-                    f"{table[header][i]:>{lenghtes[header]}}", header
+                    f"{table[header][i]:>{lengthes[header]}}", header
                 ),
                 headers,
             )
@@ -276,7 +277,7 @@ class TablePrinter:
             print(line)
 
         line_bot = f"{self.vert}{self.t_up}{self.vert}".join(
-            map(lambda header: self.vert * lenghtes[header], headers)
+            map(lambda header: self.vert * lengthes[header], headers)
         )
         line_bot = (
             self.bottom_left + self.vert + line_bot + self.vert + self.bottom_right