def run_per_instance_models()

in model/traffic/heron/prophet.py [0:0]

• 17 lines of code
• 5 McCabe index (conditional complexity)

def run_per_instance_models(models: INSTANCE_MODELS,
                            future_mins: int) -> pd.DataFrame:

    output: pd.DataFrame = None

    for spout_comp, task_dict in models.items():
        for task, stream_models in task_dict.items():
            for stream, model in stream_models.items():

                future: pd.DataFrame = model.make_future_dataframe(
                    periods=future_mins, freq='T', include_history=False)

                forecast: pd.DataFrame = model.predict(future)

                forecast["stream"] = stream
                forecast["task"] = task
                forecast["component"] = spout_comp

                if output is None:
                    output = forecast
                else:
                    output = output.append(forecast)

    return output