Merge pull request #119 from daft-dev/bugfix

fix bug when savefig() is called before render()
daft-dev · Apr 7, 2020 · 5c57191 · 5c57191
2 parents 119c323 + ce47ebe
commit 5c57191
Show file tree

Hide file tree

Showing 12 changed files with 3 additions and 0 deletions.
diff --git a/daft.py b/daft.py
@@ -105,6 +105,7 @@ def __init__(
             label_params=label_params,
             dpi=dpi,
         )
+        self.figure = None
 
     def add_node(
         self,
@@ -462,6 +463,8 @@ def savefig(self, fname, *args, **kwargs):
         """
         kwargs["bbox_inches"] = kwargs.get("bbox_inches", "tight")
         kwargs["dpi"] = kwargs.get("dpi", self._dpi)
+        if not self.figure:
+            self.render()
         self.figure.savefig(fname, *args, **kwargs)
 
 

diff --git a/docs/_static/examples/astronomy-thumb.png b/docs/_static/examples/astronomy-thumb.png
diff --git a/docs/_static/examples/classic-thumb.png b/docs/_static/examples/classic-thumb.png
diff --git a/docs/_static/examples/deconvolution-thumb.png b/docs/_static/examples/deconvolution-thumb.png
diff --git a/docs/_static/examples/exoplanets-thumb.png b/docs/_static/examples/exoplanets-thumb.png
diff --git a/docs/_static/examples/fixed-thumb.png b/docs/_static/examples/fixed-thumb.png
diff --git a/docs/_static/examples/mrf-thumb.png b/docs/_static/examples/mrf-thumb.png
diff --git a/docs/_static/examples/nocircles-thumb.png b/docs/_static/examples/nocircles-thumb.png
diff --git a/docs/_static/examples/nogray-thumb.png b/docs/_static/examples/nogray-thumb.png
diff --git a/docs/_static/examples/recursive-thumb.png b/docs/_static/examples/recursive-thumb.png
diff --git a/docs/_static/examples/weaklensing-thumb.png b/docs/_static/examples/weaklensing-thumb.png
diff --git a/docs/_static/examples/wordy-thumb.png b/docs/_static/examples/wordy-thumb.png