TNO Intern

Commit 9c6c9adc authored by Florian Knappers's avatar Florian Knappers
Browse files

detail

parent e7b744c4

src/pythermogis/workflow/utc/utc.py

+3 −4
Original line number Diff line number Diff line
@@ -12,10 +12,7 @@ def run_utc_workflow(config: UTCConfiguration) -> xr.DataTree: 
    root = xr.DataTree()

    base = Path(config.input_data_dir)



    temperature_model = xr.open_dataset(base / "NetherlandsTemperatureModel_extended.nc")

    temperature_model = temperature_model.transpose("y", "x", "z")



    chunk_size = {'x': 200, 'y': 200}

    temperature_model = xr.open_dataset(base / "NetherlandsTemperatureModel_extended.nc").transpose("y", "x", "z")





    for aquifer in config.aquifers:
@@ -215,6 +212,8 @@ def compute_results_for_aquifer( 
    for name, result in zip(output_names, results, strict=True):

        aquifer_ds[name] = result



    aquifer_ds.compute()



    print(f"Computed results for {aquifer_name}")



    return aquifer_ds, n_cells