Add publisherdb api call response check in case call fails

PiperOrigin-RevId: 715146504
GoogleCloudPlatform · Jan 14, 2025 · 7696140 · 7696140
1 parent a7c1b9a
commit 7696140
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 10 deletions.
diff --git a/notebooks/community/model_garden/model_garden_pytorch_llama3_1_deployment.ipynb b/notebooks/community/model_garden/model_garden_pytorch_llama3_1_deployment.ipynb
@@ -605,11 +605,12 @@
         "    response.raise_for_status()\n",
         "    response = response.json()\n",
         "    if (\n",
-        "        response.get(\"supportedActions\", {})\n",
-        "        .get(\"multiDeployVertex\", {})\n",
-        "        .get(\"multiDeployVertex\", {})\n",
-        "        .items()\n",
-        "        == {}.items()\n",
+        "        len(\n",
+        "            response.get(\"supportedActions\", {})\n",
+        "            .get(\"multiDeployVertex\", {})\n",
+        "            .get(\"multiDeployVertex\", {})\n",
+        "        )\n",
+        "        == 0\n",
         "    ):\n",
         "        raise ValueError(\n",
         "            f\"No supportedActions.multiDeployVertex found in {REGION}. You can skip this section or try a different region.\"\n",

diff --git a/notebooks/community/model_garden/model_garden_pytorch_llama3_2_deployment.ipynb b/notebooks/community/model_garden/model_garden_pytorch_llama3_2_deployment.ipynb
@@ -592,11 +592,12 @@
         "    response.raise_for_status()\n",
         "    response = response.json()\n",
         "    if (\n",
-        "        response.get(\"supportedActions\", {})\n",
-        "        .get(\"multiDeployVertex\", {})\n",
-        "        .get(\"multiDeployVertex\", {})\n",
-        "        .items()\n",
-        "        == {}.items()\n",
+        "        len(\n",
+        "            response.get(\"supportedActions\", {})\n",
+        "            .get(\"multiDeployVertex\", {})\n",
+        "            .get(\"multiDeployVertex\", {})\n",
+        "        )\n",
+        "        == 0\n",
         "    ):\n",
         "        raise ValueError(\n",
         "            f\"No supportedActions.multiDeployVertex found in {REGION}. You can skip this section or try a different region.\"\n",