feat: plot(show, close) options

model_tools/parameters_storage.py

@@ -28,13 +28,26 @@ class ParametersStorage(NestedMKDict):
             key = '.'.join(key)
             labels.setdefault('paths', []).append(key)
 
-    def plot(self) -> None:
+    def plot(
+        self,
+        *args,
+        close: bool=True,
+        show: bool=True,
+        **kwargs
+    ) -> None:
         from dagflow.plot import plot_auto
+        if close:
+            from matplotlib.pyplot import close
+        if show:
+            from matplotlib.pyplot import show
         for key, value in self.walkitems():
             if not isinstance(value, Output):
                 continue
 
-            plot_auto(value)
+            plot_auto(value, *args, **kwargs)
+
+            if show: show()
+            if close: close()
 
     def to_list(self, **kwargs) -> list:
         return self.visit(ParametersVisitor(kwargs)).data_list

models/dayabay_v0.py

 from dagflow.bundles.load_parameters import load_parameters
 from pathlib import Path
-from matplotlib.pyplot import show
 
 from dagflow.graph import Graph
 from dagflow.graphviz import savegraph
@@ -63,8 +62,7 @@ def model_dayabay_v0():
         outputs['edges.energy_common']= (energy_edges:=Array("energy_edges", linspace(0, 12, 241)).outputs[0])
 
     storage.read_paths()
-    storage('outputs').plot()
-    show()
+    storage('outputs').plot(close=False, show=True)
 
     storage['parameter.normalized.detector.eres.b_stat'].value = 1
     storage['parameter.normalized.detector.eres.a_nonuniform'].value = 2