Backport PR #414: Fix search filter and CLI test (#414)

Fix filter for selecting one deterministic belief per event, which was duplicating index levels. * Fix CLI test Signed-off-by: F.N. Claessen <felix@seita.nl> * Fix: do not duplicate index levels in groupby Signed-off-by: F.N. Claessen <felix@seita.nl> * Changelog entry Signed-off-by: F.N. Claessen <felix@seita.nl>
FlexMeasures · Apr 9, 2022 · 40b2a96 · 40b2a96
1 parent d5250c9
commit 40b2a96
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/documentation/changelog.rst b/documentation/changelog.rst
@@ -8,6 +8,7 @@ v0.9.2 | April XX, 2022
 Bugfixes
 --------
 * Prefer unit conversions to short stock units [see `PR #412 <http://www.github.com/FlexMeasures/flexmeasures/pull/412>`_]
+* Fix filter for selecting one deterministic belief per event, which was duplicating index levels [see `PR #414 <http://www.github.com/FlexMeasures/flexmeasures/pull/414>`_]
 
 
 v0.9.1 | March 31, 2022

diff --git a/flexmeasures/cli/tests/test_data_show.py b/flexmeasures/cli/tests/test_data_show.py
@@ -100,7 +100,7 @@ def test_plot_beliefs(app, fresh_db, setup_beliefs_fresh_db):
         [
             "--sensor-id",
             sensor.id,
-            "--from",
+            "--start",
             "2021-03-28T16:00+01",
             "--duration",
             "PT1H",

diff --git a/flexmeasures/data/models/time_series.py b/flexmeasures/data/models/time_series.py
@@ -560,7 +560,7 @@ def search(
                 # todo: move to timely-beliefs: select mean/median belief
                 bdf = (
                     bdf.for_each_belief(get_median_belief)
-                    .groupby(level=["event_start", "belief_time"])
+                    .groupby(level=["event_start", "belief_time"], group_keys=False)
                     .apply(lambda x: x.head(1))
                 )
             if resolution is not None: