@@ -153,7 +153,7 @@ def plot_traces(files, styles, width=None, depth=None, mem_unit="MB", time_unit=
out_dir = Path(__file__).resolve().parent / "MemoryPlots" / path
out_dir.mkdir(parents=True, exist_ok=True)
plt.tight_layout();
- plt.savefig(out_dir / f"{name or '|'.join(set(models)) + '_memory'}.png", dpi=300)
+ plt.savefig(out_dir / f"{name or '|'.join(set(models))}.png", dpi=300)
def main():