def prepare_data()

in src/deep_demand_forecast/monitor.py [0:0]

• 11 lines of code
• 1 McCabe index (conditional complexity)

def prepare_data(path: str):
    ds = load_multivariate_datasets(Path(path))
    train_ds = next(iter(ds.train))
    test_ds = next(iter(ds.test))
    logger.info(
        f"original train data shape {train_ds['target'].shape}, test data shape {test_ds['target'].shape}"
    )
    train_df = multivar_df(train_ds)
    test_df = multivar_df(test_ds)
    assert all(train_df.columns == test_df.columns)
    return train_df, test_df