Use split (mlflow#5522)

Signed-off-by: dbczumar <[email protected]>
Zomato · Mar 23, 2022 · 1b2793b · 1b2793b
1 parent e78d6e9
commit 1b2793b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlflow/sagemaker/__init__.py b/mlflow/sagemaker/__init__.py
@@ -2518,7 +2518,7 @@ def predict(self, deployment_name, df):
             )
             response = sage_client.invoke_endpoint(
                 EndpointName=deployment_name,
-                Body=json.dumps(_get_jsonable_obj(df)),
+                Body=json.dumps(_get_jsonable_obj(df, pandas_orient="split")),
                 ContentType="application/json",
             )