IC: Fix data sorting at the end of generation.

setup.py
Pierre-Antoine Rouby 2023-11-29 16:52:01 +01:00
parent 962bfa945d
commit 2fd44fbfb4
1 changed files with 13 additions and 0 deletions

View File

@ -395,6 +395,7 @@ class InitialConditions(SQLSubModel):
self._data.append(new)
previous_elevation = elevation
self._generate_resort_data(profiles)
def generate_discharge(self, discharge: float):
self._data = []
@ -438,3 +439,15 @@ class InitialConditions(SQLSubModel):
previous_elevation = elevation
self._data.append(new)
self._generate_resort_data(profiles)
def _generate_resort_data(self, profiles):
is_reverse = False
if profiles[0].kp > profiles[-1].kp:
is_reverse = True
self._data.sort(
reverse=not is_reverse,
key=lambda d: d['kp']
)