new plotting strategy

2025-12-19 15:50:47 +01:00
parent 68a7def5e6
commit 92104a6b06
3 changed files with 45 additions and 49 deletions
--- a/ThermalSolver/graph_datamodule.py
+++ b/ThermalSolver/graph_datamodule.py
@@ -82,7 +82,9 @@ class GraphDataModule(LightningDataModule):
        conductivity = torch.tensor(
            snapshot["conductivity"], dtype=torch.float32
        )
-        temperature = torch.tensor(snapshot["temperature"], dtype=torch.float32)[:50]
+        temperature = torch.tensor(
+            snapshot["temperature"], dtype=torch.float32
+        )[:50]

        pos = torch.tensor(geometry["points"], dtype=torch.float32)[:, :2]