Skip to content

Commit e1a5acc

Browse files
committed
style: Fix lint errors
1 parent e62534c commit e1a5acc

1 file changed

Lines changed: 19 additions & 25 deletions

File tree

uvdat/core/tasks/analytics/uncertainty_quantification.py

Lines changed: 19 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,12 @@
11
from __future__ import annotations
22

3-
import numpy as np
4-
import tempfile
5-
63
from celery import shared_task
74
from django.conf import settings
5+
import numpy as np
86

97
from uvdat.core.models import TaskResult
108

11-
from .analysis_type import AnalysisType, AnalysisTask
9+
from .analysis_type import AnalysisTask, AnalysisType
1210
from .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)
6965
def 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

Comments
 (0)