[Example]change the order of plt's code (#1052)

lijialin03 · web-flow · commit 8cea0c0c70ab · 2024-12-23T20:21:05.000+08:00
diff --git a/examples/deepcfd/deepcfd.py b/examples/deepcfd/deepcfd.py
@@ -194,8 +194,8 @@ def predict_and_save_plot(
     )
     plt.colorbar(orientation="horizontal")
     plt.tight_layout()
-    plt.show()
     plt.savefig(os.path.join(plot_dir, f"cfd_{index}.png"), bbox_inches="tight")
+    plt.show()
 
 
 def train(cfg: DictConfig):

Original file line number	Diff line number	Diff line change
`@@ -194,8 +194,8 @@ def predict_and_save_plot(`
`194`	`194`	`)`
`195`	`195`	`plt.colorbar(orientation="horizontal")`
`196`	`196`	`plt.tight_layout()`
`197`		`- plt.show()`
`198`	`197`	`plt.savefig(os.path.join(plot_dir, f"cfd_{index}.png"), bbox_inches="tight")`
	`198`	`+ plt.show()`
`199`	`199`
`200`	`200`
`201`	`201`	`def train(cfg: DictConfig):`