Fix type error

GoogleCloudPlatform · Jun 4, 2024 · ba0067e · ba0067e
1 parent 6216a5e
commit ba0067e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/notebooks/end-to-end-structured/labs/4a_sample_babyweight.ipynb b/notebooks/end-to-end-structured/labs/4a_sample_babyweight.ipynb
@@ -618,6 +618,7 @@
     "<pre style=\"color:white\">\n",
     "twins_etc = dict(zip([1,2,3,4,5],\n",
     "                   [\"Single(1)\", \"Twins(2)\", \"Triplets(3)\", \"Quadruplets(4)\", \"Quintuplets(5)\"]))\n",
+    "df[\"plurality\"] = df[\"plurality\"].apply(str)\n",
     "df[\"plurality\"].replace(twins_etc, inplace=True)\n",
     "</pre>\n",
     "Lack of ultrasound:\n",

diff --git a/notebooks/end-to-end-structured/solutions/4a_sample_babyweight.ipynb b/notebooks/end-to-end-structured/solutions/4a_sample_babyweight.ipynb
@@ -627,6 +627,7 @@
     "            ],\n",
     "        )\n",
     "    )\n",
+    "    df[\"plurality\"] = df[\"plurality\"].apply(str)\n",
     "    df[\"plurality\"].replace(twins_etc, inplace=True)\n",
     "\n",
     "    # Clone data and mask certain columns to simulate lack of ultrasound\n",