model: save trend model plots in baseline directory

Signed-off-by: U. Artie Eoff <[email protected]>

Author: uartie

lib/baseline.py

@@ -34,6 +34,8 @@ def __init__(self, filename, rebase = False):
       if os.path.isdir(self.filename): # expanded reference files
         for root, dirs, files in os.walk(self.filename):
           for name in files:
+            # ignore trendline model plots
+            if name.endswith(".svg"): continue
             with open(os.path.join(root, name), "r") as fd:
               self.references.update(json.load(fd))
       else: # flat reference file for backwards compatibility

model/encode/util.py

@@ -128,6 +128,12 @@ def fit(self):
     plt.ylim([15, 100])
     plt.legend()
     plt.savefig(get_media().artifacts.reserve("svg"))
+
+    if get_media().baseline.rebase:
+      svgdir = os.path.join(get_media().baseline.filename, "model", "encode", f"{codec}")
+      makepath(svgdir)
+      plt.savefig(os.path.join(svgdir, f"{self.case}.svg"))
+
     # plt.show()
     plt.clf()