Updated the serialisation function for CSV (aws#3580)

Fixed string formatting issue for inference
hchings · Aug 27, 2022 · e75876c · e75876c
1 parent 6ccaf6c
commit e75876c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...lar/train-register-deploy-pipeline-model/train register and deploy a pipeline model.ipynb b/...lar/train-register-deploy-pipeline-model/train register and deploy a pipeline model.ipynb
@@ -1130,7 +1130,7 @@
     "data = data.drop(\"medianHouseValue\", axis=1)\n",
     "\n",
     "pred_count = 10\n",
-    "payload = data.iloc[:pred_count].to_string(header=False, index=False).replace(\"  \", \",\")\n",
+    "payload = data.iloc[:pred_count].to_csv(header=False, index=False)\n",
     "p = predictor.predict(payload, initial_args={\"ContentType\": \"text/csv\"})\n",
     "print(p.decode(\"utf-8\"))"
    ]
@@ -1231,4 +1231,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
+}