diff --git a/courses/machine_learning/deepdive2/production_ml/babyweight/train_deploy.ipynb b/courses/machine_learning/deepdive2/production_ml/babyweight/train_deploy.ipynb
index ec2ce4b0c7..5379dc3b6a 100644
--- a/courses/machine_learning/deepdive2/production_ml/babyweight/train_deploy.ipynb
+++ b/courses/machine_learning/deepdive2/production_ml/babyweight/train_deploy.ipynb
@@ -38,42 +38,11 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Requirement already satisfied: google-cloud-bigquery==2.26.0 in /opt/conda/lib/python3.7/site-packages (2.26.0)\n",
-      "Requirement already satisfied: grpcio<2.0dev,>=1.38.1 in /opt/conda/lib/python3.7/site-packages (from google-cloud-bigquery==2.26.0) (1.38.1)\n",
-      "Requirement already satisfied: proto-plus>=1.10.0 in /opt/conda/lib/python3.7/site-packages (from google-cloud-bigquery==2.26.0) (1.19.0)\n",
-      "Requirement already satisfied: google-api-core[grpc]<3.0.0dev,>=1.29.0 in /opt/conda/lib/python3.7/site-packages (from google-cloud-bigquery==2.26.0) (1.31.2)\n",
-      "Requirement already satisfied: protobuf>=3.12.0 in /opt/conda/lib/python3.7/site-packages (from google-cloud-bigquery==2.26.0) (3.18.0)\n",
-      "Requirement already satisfied: google-cloud-core<3.0.0dev,>=1.4.1 in /opt/conda/lib/python3.7/site-packages (from google-cloud-bigquery==2.26.0) (2.0.0)\n",
-      "Requirement already satisfied: google-resumable-media<3.0dev,>=0.6.0 in /opt/conda/lib/python3.7/site-packages (from google-cloud-bigquery==2.26.0) (2.0.3)\n",
-      "Requirement already satisfied: packaging>=14.3 in /opt/conda/lib/python3.7/site-packages (from google-cloud-bigquery==2.26.0) (21.0)\n",
-      "Requirement already satisfied: requests<3.0.0dev,>=2.18.0 in /opt/conda/lib/python3.7/site-packages (from google-cloud-bigquery==2.26.0) (2.25.1)\n",
-      "Requirement already satisfied: six>=1.13.0 in /opt/conda/lib/python3.7/site-packages (from google-api-core[grpc]<3.0.0dev,>=1.29.0->google-cloud-bigquery==2.26.0) (1.16.0)\n",
-      "Requirement already satisfied: google-auth<2.0dev,>=1.25.0 in /opt/conda/lib/python3.7/site-packages (from google-api-core[grpc]<3.0.0dev,>=1.29.0->google-cloud-bigquery==2.26.0) (1.35.0)\n",
-      "Requirement already satisfied: setuptools>=40.3.0 in /opt/conda/lib/python3.7/site-packages (from google-api-core[grpc]<3.0.0dev,>=1.29.0->google-cloud-bigquery==2.26.0) (58.0.4)\n",
-      "Requirement already satisfied: pytz in /opt/conda/lib/python3.7/site-packages (from google-api-core[grpc]<3.0.0dev,>=1.29.0->google-cloud-bigquery==2.26.0) (2021.1)\n",
-      "Requirement already satisfied: googleapis-common-protos<2.0dev,>=1.6.0 in /opt/conda/lib/python3.7/site-packages (from google-api-core[grpc]<3.0.0dev,>=1.29.0->google-cloud-bigquery==2.26.0) (1.53.0)\n",
-      "Requirement already satisfied: cachetools<5.0,>=2.0.0 in /opt/conda/lib/python3.7/site-packages (from google-auth<2.0dev,>=1.25.0->google-api-core[grpc]<3.0.0dev,>=1.29.0->google-cloud-bigquery==2.26.0) (4.2.2)\n",
-      "Requirement already satisfied: pyasn1-modules>=0.2.1 in /opt/conda/lib/python3.7/site-packages (from google-auth<2.0dev,>=1.25.0->google-api-core[grpc]<3.0.0dev,>=1.29.0->google-cloud-bigquery==2.26.0) (0.2.7)\n",
-      "Requirement already satisfied: rsa<5,>=3.1.4 in /opt/conda/lib/python3.7/site-packages (from google-auth<2.0dev,>=1.25.0->google-api-core[grpc]<3.0.0dev,>=1.29.0->google-cloud-bigquery==2.26.0) (4.7.2)\n",
-      "Requirement already satisfied: google-crc32c<2.0dev,>=1.0 in /opt/conda/lib/python3.7/site-packages (from google-resumable-media<3.0dev,>=0.6.0->google-cloud-bigquery==2.26.0) (1.2.0)\n",
-      "Requirement already satisfied: pyparsing>=2.0.2 in /opt/conda/lib/python3.7/site-packages (from packaging>=14.3->google-cloud-bigquery==2.26.0) (2.4.7)\n",
-      "Requirement already satisfied: pyasn1<0.5.0,>=0.4.6 in /opt/conda/lib/python3.7/site-packages (from pyasn1-modules>=0.2.1->google-auth<2.0dev,>=1.25.0->google-api-core[grpc]<3.0.0dev,>=1.29.0->google-cloud-bigquery==2.26.0) (0.4.8)\n",
-      "Requirement already satisfied: urllib3<1.27,>=1.21.1 in /opt/conda/lib/python3.7/site-packages (from requests<3.0.0dev,>=2.18.0->google-cloud-bigquery==2.26.0) (1.26.6)\n",
-      "Requirement already satisfied: certifi>=2017.4.17 in /opt/conda/lib/python3.7/site-packages (from requests<3.0.0dev,>=2.18.0->google-cloud-bigquery==2.26.0) (2021.5.30)\n",
-      "Requirement already satisfied: idna<3,>=2.5 in /opt/conda/lib/python3.7/site-packages (from requests<3.0.0dev,>=2.18.0->google-cloud-bigquery==2.26.0) (2.10)\n",
-      "Requirement already satisfied: chardet<5,>=3.0.2 in /opt/conda/lib/python3.7/site-packages (from requests<3.0.0dev,>=2.18.0->google-cloud-bigquery==2.26.0) (4.0.0)\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
-    "!pip install --user google-cloud-bigquery==2.26.0"
+    "!pip install --user google-cloud-bigquery>=2.26.0"
    ]
   },
   {
@@ -82,7 +51,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install -U google-cloud-aiplatform \"shapely<2\""
+    "!pip install -U google-cloud-aiplatform \"shapely>=2\""
    ]
   },
   {
@@ -151,7 +120,7 @@
     "os.environ[\"PROJECT\"] = PROJECT\n",
     "os.environ[\"BUCKET\"] = BUCKET\n",
     "os.environ[\"REGION\"] = REGION\n",
-    "os.environ[\"TFVERSION\"] = \"2.6\"\n",
+    "os.environ[\"TFVERSION\"] = \"2.11\"\n",
     "os.environ[\"PYTHONVERSION\"] = \"3.7\""
    ]
   },