diff --git a/src/handlers/predict_sklearn.py b/src/handlers/predict_sklearn.py index 71a918a..7f7728c 100644 --- a/src/handlers/predict_sklearn.py +++ b/src/handlers/predict_sklearn.py @@ -31,7 +31,7 @@ def sklearn_post_handler(request: PredictionRequestPydantic): else prediction[jaqpot_row_id, i] for i, feature in enumerate(request.model["dependentFeatures"]) } - results["jaqpotInternalMetadata"] = { + results["jaqpotMetadata"] = { "AD": None, "probabilities": probabilities[jaqpot_row_id], "jaqpotRowId": jaqpot_row_id