fix: in case of a tied rank, where multiple sources have exactly the …

…same number of data points, arbitrarily pick the first one in the data Signed-off-by: F.N. Claessen <felix@seita.nl>
FlexMeasures · Aug 1, 2023 · 12bac09 · 12bac09
1 parent 7d13ee3
commit 12bac09
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/flexmeasures/data/models/charts/belief_charts.py b/flexmeasures/data/models/charts/belief_charts.py
@@ -191,6 +191,17 @@ def daily_heatmap(
                         "frame": [None, None],
                     },
                     {"filter": "datum.source_rank == 1"},
+                    # In case of a tied rank, arbitrarily choose the first one occurring in the data
+                    {
+                        "window": [
+                            {
+                                "op": "first_value",
+                                "field": "source.id",
+                                "as": "first_source_id",
+                            }
+                        ],
+                    },
+                    {"filter": "datum.source.id == datum.first_source_id"},
                 ],
             },
             {