added IPython rich display to Figure class

btel · Jan 23, 2018 · d909b5b · d909b5b
1 parent 71bc73a
commit d909b5b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/svgutils/compose.py b/src/svgutils/compose.py
@@ -268,12 +268,15 @@ def save(self, fname):
         element.save(os.path.join(CONFIG['figure.save_path'], fname))
 
     def tostr(self):
-        """Export SVG as string"""
+        """Export SVG as a string"""
         element = _transform.SVGFigure(self.width, self.height)
         element.append(self)
         svgstr = element.to_str()
         return svgstr
 
+    def _repr_svg_(self):
+        return self.tostr().decode('ascii')
+
     def tile(self, ncols, nrows):
         """Automatically tile the panels of the figure.