Add more convenient function to save graph.

Author: Matthieu Ancellin

labelled_functions/abstract.py

@@ -173,7 +173,7 @@ def apply_in_namespace(self, namespace: Union[Dict[str, Any], xr.Dataset]) -> Un
     def _graph(self, **kwargs):
         pass
 
-    def graph(self, backend='graphviz', rankdir='LR'):
+    def graph(self, backend='graphviz', rankdir='TB'):
         inputs_nodes, default_values, output_nodes, function_nodes, dummy_nodes, edges = self._graph()
 
         if backend == 'graphviz':
@@ -227,3 +227,9 @@ def format_node_label(name, value):
 
         return G
 
+    def save_graph(self, filepath=None):
+        from pathlib import Path
+        if filepath is None:
+            filepath = Path.cwd() / (self.name + ".pdf")
+        self.graph(backend='pygraphviz', rankdir='TB').draw(str(filepath), prog='dot')
+