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)