Drop belief_time, source and cumulative_probability

Signed-off-by: Victor Garcia Reolid <victor@seita.nl>

flexmeasures/data/models/reporting/pandas_reporter.py

@@ -43,6 +43,8 @@ def fetch_data(
         Fetches the time_beliefs from the database
         """
 
+        droplevels = self._config.get("droplevels", False)
+
         self.data = {}
         for input_search_parameters in input:
             _input_search_parameters = input_search_parameters.copy()
@@ -71,6 +73,9 @@ def fetch_data(
             for source in bdf.sources.unique():
                 self.data[f"source_{source.id}"] = source
 
+            if droplevels:
+                bdf = bdf.droplevel([1,2,3])
+
             # store BeliefsDataFrame as local variable
             self.data[name] = bdf
 

flexmeasures/data/schemas/reporting/pandas_reporter.py

@@ -88,6 +88,8 @@ class PandasReporterConfigSchema(ReporterConfigSchema):
     )
     transformations = fields.List(fields.Nested(PandasMethodCall()), required=True)
 
+    droplevels = fields.Bool(required=False, default=False)
+
     @validates_schema
     def validate_chaining(self, data, **kwargs):
         """