diff --git a/mlflow/metrics/genai/genai_metric.py b/mlflow/metrics/genai/genai_metric.py index 257f1f30cf29f..769d93abd3258 100644 --- a/mlflow/metrics/genai/genai_metric.py +++ b/mlflow/metrics/genai/genai_metric.py @@ -281,6 +281,7 @@ def score_model_on_one_payload( if e.error_code in [ ErrorCode.Name(BAD_REQUEST), ErrorCode.Name(UNAUTHENTICATED), + ErrorCode.Name(INVALID_PARAMETER_VALUE) ]: raise MlflowException(e) return None, f"Failed to score model on payload. Error: {e!s}"