Updated the plot inclusion.

transientlunatic · Dec 23, 2015 · 2c3ffb5 · 2c3ffb5
1 parent 2708966
commit 2c3ffb5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/otter/otter.py b/otter/otter.py
@@ -93,10 +93,13 @@ def write_image(self, url):
         """.format(url)
         self._write(html_str)
 
-    def write_plot(self, filename=None):
+    def write_plot(self, figure=None, filename=None):
         if filename==None:
             filename = "{}.png".format(uuid.uuid4().hex)
-        plt.savefig(self.reportfolder+"/"+filename, dpi=300)
+	if figure:
+        	figure.savefig(self.reportfolder+"/"+filename, dpi=300)
+	else:
+		plt.savefig(self.reportfolder+"/"+filename, dpi=300)
         self.write_image(self.foldername+filename)
 
 
diff --git a/setup.py b/setup.py
@@ -24,7 +24,7 @@
 
 setup(
     name='otter-report',
-    version='0.0.3',
+    version='0.0.4',
     description="The simple HTML report generator for Python jobs.",
     long_description=readme + '\n\n' + history,
     author="Daniel Williams",