fix model

ThermalSolver/graph_module.py

@@ -7,6 +7,7 @@ from matplotlib.tri import Triangulation
 from .model.finite_difference import FiniteDifferenceStep
 import os
 
+
 def import_class(class_path: str):
     module_path, class_name = class_path.rsplit(".", 1)  # split last dot
     module = importlib.import_module(module_path)  # import the module
@@ -43,6 +44,7 @@ def _plot_mesh(pos, y, y_pred, batch, i, batch_idx):
     plt.savefig(name, dpi=72)
     plt.close()
 
+
 def _plot_losses(losses, batch_idx):
     folder = f"{batch_idx:02d}_images"
     plt.figure()