From 5ba0d5fcc03d56d35da42e8a03b94154692de8c0 Mon Sep 17 00:00:00 2001
From: bdattoma <bdattoma@redhat.com>
Date: Tue, 9 Jan 2024 11:11:02 +0100
Subject: [PATCH] disable warnings in jupyter

---
 .../CLI/ModelServing/caikit-py-query.ipynb    | 24 +++++++++++++++++--
 1 file changed, 22 insertions(+), 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
index da3367692..f67cfbe50 100644
--- 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()"