from telemetry_quality import repeat_daily_profile
import datetime
from methods.common import save_model


def train(data_obj):
    aid = data_obj["config"]["anlagen_id"]
    return {"trained": True, "path": save_model(aid, 10, {"type": "repeat_pv_24h"})}


def predict(data_obj):
    return repeat_daily_profile(data_obj["df_hist"], data_obj["df_fut"].index, 'PV')