test_graph

relari-ai · Feb 19, 2024 · 0b37db1 · 0b37db1
1 parent a7eb1a3
commit 0b37db1
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/continuous_eval/eval/manager.py b/continuous_eval/eval/manager.py
@@ -207,5 +207,14 @@ def load_test_results(self, filepath: Path):
         with open(filepath, "r") as json_file:
             self._test_results = json.load(json_file)
 
+    def test_graph(self):
+        pipeline_graph = self._pipeline.graph_repr()
+        tests = "\n    %% Tests\n"
+        for module, results in self._test_results.items():
+            metrics = "<br>".join([f"<i>{metric}: {value}</i>" for metric, value in results.items()])
+            tests += f"    {module}[<b>{module}</b><br>---<br>{metrics}]\n"
+
+        return pipeline_graph + tests
+
 
 eval_manager = EvaluationManager()