disable warnings in jupyter

bdattoma · Jan 9, 2024 · 5ba0d5f · 5ba0d5f
1 parent 72b355a
commit 5ba0d5f
Showing 1 changed file with 22 additions and 2 deletions.
diff --git a/ods_ci/tests/Resources/CLI/ModelServing/caikit-py-query.ipynb b/ods_ci/tests/Resources/CLI/ModelServing/caikit-py-query.ipynb
@@ -85,8 +85,18 @@
    },
    "outputs": [],
    "source": [
+    "%%capture --no-display\n",
     "http_client = HttpClient(f\"{HTTP_HOST}:{PORT}\", verify=False)\n",
-    "generated_text = http_client.generate_text(MODEL_ID, QUERY_TEXT)\n",
+    "generated_text = http_client.generate_text(MODEL_ID, QUERY_TEXT)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "b134f83a",
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "print(generated_text)\n",
     "assert generated_text.strip() == EXPECTED_ANSWER.strip()"
    ]
@@ -186,7 +196,17 @@
    },
    "outputs": [],
    "source": [
-    "grpc_client = GrpcClient(GRPC_HOST, int(PORT), verify=False)\n",
+    "%%capture --no-display\n",
+    "grpc_client = GrpcClient(GRPC_HOST, int(PORT), verify=False)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "3bed4431",
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "generated_text = grpc_client.generate_text(MODEL_ID, QUERY_TEXT)\n",
     "print(generated_text)\n",
     "assert generated_text.strip() == EXPECTED_ANSWER.strip()"