add save png

Author: kunwuz

causallearn/graph/GraphClass.py

@@ -193,10 +193,9 @@ def draw_pydot_graph(self):
         warnings.filterwarnings("ignore", category=UserWarning)
         pyd = GraphUtils.to_pydot(self.G)
         tmp_png = pyd.create_png(f="png")
+        pyd.write_png("result.png")
         fp = io.BytesIO(tmp_png)
         img = mpimg.imread(fp, format='png')
         plt.axis('off')
         plt.imshow(img)
         plt.show()
-
-

setup.py

@@ -3,7 +3,7 @@
 with open('README.md', 'r') as fh:
     README = fh.read()
 
-VERSION = '0.1.1.8'
+VERSION = '0.1.1.9'
 
 setuptools.setup(
     name='causal-learn',