Merge branch 'main' into model_monitoring_container

fix: tf serving workaround (#917 )
* fix: pin TF serving image * fix: pin TF serving image Co-authored-by: gericdong <itseric@google.com>
2026-09-28 16:11:56 +00:00 · 2022-09-01 10:20:20 -07:00 · 2022-09-01 12:58:27 -04:00 · 2022-09-01 15:26:45 +00:00 · 2022-09-01 15:25:40 +00:00
2 changed files with 1878 additions and 6 deletions
@@ -702,7 +702,7 @@
        "        + f\"/{PRIVATE_REPO}\"\n",
        "        + \"/tf_serving:gpu\"\n",
        "    )\n",
-        "    TF_IMAGE = \"tensorflow/serving:latest-gpu\"\n",
+        "    TF_IMAGE = \"tensorflow/serving:2.5.4-gpu\"\n",
        "else:\n",
        "    DEPLOY_IMAGE = (\n",
        "        f\"{REGION}-docker.pkg.dev/\"\n",
@@ -710,15 +710,15 @@
        "        + f\"/{PRIVATE_REPO}\"\n",
        "        + \"/tf_serving:cpu\"\n",
        "    )\n",
-        "    TF_IMAGE = \"tensorflow/serving:latest\"\n",
+        "    TF_IMAGE = \"tensorflow/serving:2.5.4\"\n",
        "\n",
        "if not IS_COLAB:\n",
        "    if DEPLOY_GPU:\n",
-        "        ! sudo docker pull tensorflow/serving:latest-gpu\n",
+        "        ! sudo docker pull tensorflow/serving:2.5.4-gpu\n",
        "    else:\n",
-        "        ! sudo docker pull tensorflow/serving:latest\n",
+        "        ! sudo docker pull tensorflow/serving:2.5.4\n",
        "\n",
-        "    ! docker tag tensorflow/serving $DEPLOY_IMAGE\n",
+        "    ! docker tag $TF_IMAGE $DEPLOY_IMAGE\n",
        "    ! docker push $DEPLOY_IMAGE\n",
        "else:\n",
        "    # install docker daemon\n",
@@ -1434,7 +1434,7 @@
      },
      "outputs": [],
      "source": [
-        "delete_bucket = False\n",
+        "delete_bucket = True\n",
        "delete_model = True\n",
        "delete_endpoint = True\n",
        "delete_batch_job = True\n",
Author	SHA1	Message	Date
Andrew FerlitschandGitHub	5a1285d1a3	Merge branch 'main' into model_monitoring_container	2022-09-01 10:20:20 -07:00
Andrew Ferlitsch GitHub gericdong	6ddcaa540a	fix: tf serving workaround (#917 ) * fix: pin TF serving image * fix: pin TF serving image Co-authored-by: gericdong <itseric@google.com>	2022-09-01 12:58:27 -04:00
Andrew Ferlitsch	26fad6b647	feat: add notebook using custom deployment container	2022-09-01 15:26:45 +00:00
Andrew Ferlitsch	779e695dd6	feat: add notebook using custom deployment container	2022-09-01 15:25:40 +00:00