From 7feba10c9d2c1a4aff24aa8b702f4c215e8f4968 Mon Sep 17 00:00:00 2001 From: Andrei Batomunkuev Date: Fri, 12 Nov 2021 15:35:01 -0500 Subject: [PATCH] Updated assert statement, changed from MALFORMED_REQUEST to BAD_REQUEST --- tests/models/test_cli.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/models/test_cli.py b/tests/models/test_cli.py index a0e858aca83d1..c5711826d96af 100644 --- a/tests/models/test_cli.py +++ b/tests/models/test_cli.py @@ -31,7 +31,7 @@ get_safe_port, pyfunc_serve_and_score_model, ) -from mlflow.protos.databricks_pb2 import ErrorCode, MALFORMED_REQUEST +from mlflow.protos.databricks_pb2 import ErrorCode, BAD_REQUEST from mlflow.pyfunc.scoring_server import ( CONTENT_TYPE_JSON_SPLIT_ORIENTED, CONTENT_TYPE_JSON, @@ -458,6 +458,6 @@ def _validate_with_rest_endpoint(scoring_proc, host_port, df, x, sk_model): ) scoring_response_dict = json.loads(scoring_response.content) assert "error_code" in scoring_response_dict - assert scoring_response_dict["error_code"] == ErrorCode.Name(MALFORMED_REQUEST) + assert scoring_response_dict["error_code"] == ErrorCode.Name(BAD_REQUEST) assert "message" in scoring_response_dict assert "stack_trace" in scoring_response_dict