Get baseline run

conbench · Apr 22, 2021 · c2413cd · c2413cd
1 parent 297dd1a
commit c2413cd
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/conbench/app/benchmarks.py b/conbench/app/benchmarks.py
@@ -61,6 +61,20 @@ def get_display_run(self, run_id):
         self._augment(run)
         return run
 
+    def get_display_baseline_run(self, sha, machine_id):
+        response = self.api_get("api.runs", sha=sha, machine_id=machine_id)
+        if response.status_code != 200:
+            self.flash("Error getting run.")
+            return None
+
+        runs = response.json
+        if runs:
+            run = runs[0]
+            self._augment(run)
+            return run
+
+        return None
+
     def get_display_runs(self):
         runs, response = self._get_runs()
         if response.status_code != 200:

diff --git a/conbench/app/runs.py b/conbench/app/runs.py
@@ -26,7 +26,9 @@ def get(self, run_id):
         contender_run, baseline_run = self.get_display_run(run_id), None
         if contender_run:
             parent = contender_run["commit"]["parent_sha"]
-            baseline_run = self.get_display_run(run_id)  # TODO: get baseline run
+            baseline_run = self.get_display_baseline_run(
+                parent, contender_run["machine"]["id"]
+            )
 
         benchmarks, response = self._get_benchmarks(run_id)
         if response.status_code != 200: