11from __future__ import annotations
22
3- import numpy as np
4- import tempfile
5-
63from celery import shared_task
74from django .conf import settings
5+ import numpy as np
86
97from uvdat .core .models import TaskResult
108
11- from .analysis_type import AnalysisType , AnalysisTask
9+ from .analysis_type import AnalysisTask , AnalysisType
1210from .flood_simulation import FloodSimulation
1311
1412
@@ -63,8 +61,6 @@ def finalize(self, result):
6361 pass
6462
6563
66-
67-
6864@shared_task (base = AnalysisTask )
6965def uncertainty_quantification (result_id ):
7066 result = TaskResult .objects .get (id = result_id )
@@ -77,22 +73,19 @@ def uncertainty_quantification(result_id):
7773
7874 # Update name
7975 result .name = (
80- f"Uncertainty Quantification for Flood Results { flood_sim_1 .id } , { flood_sim_2 .id } , { flood_sim_3 .id } "
76+ "Uncertainty Quantification for Flood Results "
77+ f"{ flood_sim_1 .id } , { flood_sim_2 .id } , { flood_sim_3 .id } "
8178 )
8279 result .save ()
8380
84-
8581 precip1 = flood_sim_1 .outputs .get ("precipitation_level_mm" )
8682 discharge1 = flood_sim_1 .outputs .get ("discharge_ft3_per_second" )
8783 precip2 = flood_sim_2 .outputs .get ("precipitation_level_mm" )
8884 discharge2 = flood_sim_2 .outputs .get ("discharge_ft3_per_second" )
8985 precip3 = flood_sim_3 .outputs .get ("precipitation_level_mm" )
9086 discharge3 = flood_sim_3 .outputs .get ("discharge_ft3_per_second" )
9187
92-
93- result .write_status (
94- f"Calculating uncertainty..."
95- )
88+ result .write_status ("Calculating uncertainty..." )
9689
9790 precip_mean = np .mean ([precip1 , precip2 , precip3 ])
9891 precip_stde = np .std ([precip1 , precip2 , precip3 ])
@@ -105,18 +98,19 @@ def uncertainty_quantification(result_id):
10598 discharge_min = np .min ([discharge1 , discharge2 , discharge3 ])
10699 discharge_range = discharge_max - discharge_min
107100
108-
109101 result .write_status ("Saving result to database" )
110102
111- result .write_outputs ({
112- "mean_precipitation_level_mm" : precip_mean ,
113- "standard_error_precipitation_level_mm" : precip_stde ,
114- "min_precipitation_level_mm" : precip_min ,
115- "max_precipitation_level_mm" : precip_max ,
116- "range_precipitation_level_mm" : precip_range ,
117- "mean_discharge_ft3_per_second" : discharge_mean ,
118- "standard_error_discharge_ft3_per_second" : discharge_stde ,
119- "min_discharge_ft3_per_second" : discharge_min ,
120- "max_discharge_ft3_per_second" : discharge_max ,
121- "range_discharge_ft3_per_second" : discharge_range ,
122- })
103+ result .write_outputs (
104+ {
105+ "mean_precipitation_level_mm" : precip_mean ,
106+ "standard_error_precipitation_level_mm" : precip_stde ,
107+ "min_precipitation_level_mm" : precip_min ,
108+ "max_precipitation_level_mm" : precip_max ,
109+ "range_precipitation_level_mm" : precip_range ,
110+ "mean_discharge_ft3_per_second" : discharge_mean ,
111+ "standard_error_discharge_ft3_per_second" : discharge_stde ,
112+ "min_discharge_ft3_per_second" : discharge_min ,
113+ "max_discharge_ft3_per_second" : discharge_max ,
114+ "range_discharge_ft3_per_second" : discharge_range ,
115+ }
116+ )
0 commit comments