From a25c117bdbe6ba4f531f8e119f72cec1120e61d4 Mon Sep 17 00:00:00 2001
From: David Bau <david.bau@gmail.com>
Date: Sat, 19 Mar 2022 09:37:36 -0400
Subject: [PATCH] Add tight bbox.

---
 torchkit/labwidget.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchkit/labwidget.py b/torchkit/labwidget.py
index c0f474f..5af0732 100644
--- a/torchkit/labwidget.py
+++ b/torchkit/labwidget.py
@@ -981,7 +981,7 @@ class Image(Widget):
         if hasattr(obj, 'save'): # Like a PIL.Image.Image
             obj.save(buf, format='png')
         elif hasattr(obj, 'savefig'): # Like a matplotlib.figure.Figure
-            obj.savefig(buf, format='png')
+            obj.savefig(buf, format='png', bbox_inches='tight')
         self.src= 'data:image/png;base64,' + (
             base64.b64encode(buf.getvalue()).decode('utf-8'))
         buf.close()