in these lines https://github.com/OSSFE/hackathon-2026/blob/903a3b75e12701d1595ead73b16f3f9b1671e3b9/2-solve-heat-equation.py#L76C1-L77C26 @RemDelaporteMathurin