def load_raw_covariates(self, scenario: str, draw_id: int):
covariates = self.forecast_marshall.load(
key=MKeys.forecast_raw_covariates(scenario=scenario,
draw_id=draw_id))
covariates['date'] = pd.to_datetime(covariates['date'])
return covariates
def save_raw_covariates(self, covariates: pd.DataFrame, scenario: str,
draw_id: int, strict: bool):
self.forecast_marshall.dump(covariates,
key=MKeys.forecast_raw_covariates(
scenario=scenario, draw_id=draw_id),
strict=strict)