mirror of
https://github.com/Akkudoktor-EOS/EOS.git
synced 2025-04-19 08:55:15 +00:00
27 lines
849 B
Python
27 lines
849 B
Python
"""Load forecast module for load predictions."""
|
|
|
|
from typing import Optional, Union
|
|
|
|
from pydantic import Field
|
|
|
|
from akkudoktoreos.config.configabc import SettingsBaseModel
|
|
from akkudoktoreos.core.logging import get_logger
|
|
from akkudoktoreos.prediction.loadakkudoktor import LoadAkkudoktorCommonSettings
|
|
from akkudoktoreos.prediction.loadimport import LoadImportCommonSettings
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
class LoadCommonSettings(SettingsBaseModel):
|
|
"""Load Prediction Configuration."""
|
|
|
|
provider: Optional[str] = Field(
|
|
default=None,
|
|
description="Load provider id of provider to be used.",
|
|
examples=["LoadAkkudoktor"],
|
|
)
|
|
|
|
provider_settings: Optional[Union[LoadAkkudoktorCommonSettings, LoadImportCommonSettings]] = (
|
|
Field(default=None, description="Provider settings", examples=[None])
|
|
)
|