Files
thermal-conduction-ml/ThermalSolver/data_module.py