feat: heatmap shows only data from a single source

Signed-off-by: F.N. Claessen <felix@seita.nl>
FlexMeasures · Jul 31, 2023 · ef4563e · ef4563e
1 parent 8222cf8
commit ef4563e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/flexmeasures/data/models/charts/belief_charts.py b/flexmeasures/data/models/charts/belief_charts.py
@@ -178,6 +178,12 @@ def daily_heatmap(
                         "calculate": "datum.source.name + ' (ID: ' + datum.source.id + ')'",
                         "as": "source_name_and_id",
                     },
+                    # In case of multiple sources, arbitrarily choose the first one occurring in the data
+                    {
+                        "window": [{"op": "first_value", "field": "source.id", "as": "first_source_id"}],
+                        "groupby": [],
+                    },
+                    {"filter": "datum.source.id == datum.first_source_id"},
                 ],
             },
             {