def run()

in diabetes_regression/scoring/parallel_batchscore.py [0:0]

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

def run(mini_batch: pd.DataFrame) -> pd.DataFrame:
    """
    The run method is called multiple times by the runtime. Each time
    a mini-batch consisting of a portion of the input data is passed
    in as a pandas DataFrame. The run method should return the scoring
    results as a List or a pandas DataFrame.

    :param mini_batch: Dataframe containing a portion of the scoring data

    :returns: array containing the scores.
    """

    try:
        result = None

        for _, sample in mini_batch.iterrows():
            # prediction
            pred = model.predict(sample.values.reshape(1, -1))
            result = (
                np.array(pred) if result is None else np.vstack((result, pred))
            )  # NOQA: E501

        return (
            []
            if result is None
            else mini_batch.join(pd.DataFrame(result, columns=["score"]))
        )

    except Exception as ex:
        print(ex)