From 458d0a811d9edb88a2827f3700b0de2b1c4f5f81 Mon Sep 17 00:00:00 2001
From: Frank Sauerburger <f.sauerburger@cern.ch>
Date: Thu, 12 Nov 2020 19:01:20 +0100
Subject: [PATCH] Return matplotlib objects
---
nnfwtbn/plot.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/nnfwtbn/plot.py b/nnfwtbn/plot.py
index 2cd5197..6c9fa64 100644
--- a/nnfwtbn/plot.py
+++ b/nnfwtbn/plot.py
@@ -319,7 +319,7 @@ def hist(dataframe, variable, bins, stacks, selection=None,
if return_uhepp:
return uhepp_obj
else:
- uhepp_obj.render()
+ return uhepp_obj.render()
def _transpose(array):
"""Return a transposed version of the array"""
--
GitLab