IC: Fix data sorting at the end of generation.

This commit is contained in:
Pierre-Antoine Rouby committed 2023-11-29 16:52:01 +01:00
1 parent 962bfa945d
commit 2fd44fbfb4
1 file changed
+13
@@ -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']
)