FlexMeasures / flexmeasures / 29915579583

83%

push

github

Add annotation regressors to forecaster config, merged holidays command, and timezone fix (#2178)

* cli/data_add: add --timezone to add_holidays; add holidays-by-package command; add future-annotation-regressors to pipeline schema

Context:
- add_holidays stored UTC midnight timestamps, causing holidays to appear
  at wrong local time (e.g. 1AM in Amsterdam) in the UI
- No support for the 'holidays' Python package which covers more countries
  and school holidays not in workalendar
- No way to pass annotation-based binary regressors to the forecasting pipeline

Change:
- Add --timezone option to 'flexmeasures add holidays'; holidays are now
  localized to the given timezone (UTC when omitted, with a warning)
- Add new 'flexmeasures add holidays-by-package' command backed by the
  'holidays' package; supports --country, --subdiv, --category, --timezone
- Add 'holidays>=0.57' to pyproject.toml direct dependencies
- Add future_annotation_regressors field (data_key: future-annotation-regressors)
  to TrainPredictPipelineConfigSchema for annotation-based binary regressors

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Co-authored-by: Flix6x <30658763+Flix6x@users.noreply.github.com>

* cli/data_add: improve error message for invalid subdiv/category in holidays-by-package

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Co-authored-by: Flix6x <30658763+Flix6x@users.noreply.github.com>

* forecasting: add annotation regressor support as future covariates

Context:
- TrainPredictPipelineConfigSchema already supports future_annotation_regressors
- Need to wire this config field through the pipeline execution

Change:
- Add query_account_annotations() to data/queries/annotations.py
- Add _AnnotationRegressorProxy class to base pipeline for reuse in detect_and_fill_missing_values
- Add annotation_regressors parameter to BasePipeline.__init__ with proxy/name construction
- Add _load_annotation_regressor_df() to build... (continued)

236 of 276 new or added lines in 8 files covered. (85.51%)

15656 of 18994 relevant lines covered (82.43%)

0.82 hits per line