Updated assert statement, changed from MALFORMED_REQUEST to BAD_REQUEST

mlflow · Nov 12, 2021 · 7feba10 · 7feba10
1 parent 6c4cedf
commit 7feba10
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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