diff --git a/plotter.py b/plotter.py index bbfed0b..df9a782 100644 --- a/plotter.py +++ b/plotter.py @@ -72,9 +72,9 @@ class PlotRule(Rule): self.postproc.save = save return self.process_fn(arg, **kwargs) - def datafile(self, arg): + def datafile(self, name, arg): return Datafile( - name=self.description + "_" + arg, + name=name + "_" + arg, description=self.description + " ({})".format(arg), ) @@ -327,7 +327,7 @@ class Plotter(Aggregator, BaseProcessor): save.close() # Save in astrophysix format - df = rule.datafile(arg) + df = rule.datafile(name, arg) df[filetype] = plot_filename if plot_info is not None: df.plot_info = plot_info