From dc1437e5dab497ebbb20002dab6a6f13c91ddcc7 Mon Sep 17 00:00:00 2001
From: lukas <lukasvalentin.graf@usys.ethz.ch>
Date: Mon, 21 Mar 2022 10:40:24 +0100
Subject: [PATCH] returning figure object to calling program

---
 pretty_confusion_matrix/pretty_confusion_matrix.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pretty_confusion_matrix/pretty_confusion_matrix.py b/pretty_confusion_matrix/pretty_confusion_matrix.py
index 75e8b62..1c52f04 100644
--- a/pretty_confusion_matrix/pretty_confusion_matrix.py
+++ b/pretty_confusion_matrix/pretty_confusion_matrix.py
@@ -240,6 +240,8 @@ def pp_matrix(
     plt.tight_layout()  # set layout slim
     plt.show()
 
+    return fig
+
 
 def pp_matrix_from_data(
     y_test,