fix: Remove dependency on colab from prompt optimizer. (GoogleCloudPl…

…atform#1241) Co-authored-by: hootan <[email protected]> Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com> Co-authored-by: Holt Skinner <[email protected]>
Abhishekbhagwat · Oct 11, 2024 · fde9bee · fde9bee
1 parent 36a0967
commit fde9bee
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 5 deletions.
diff --git a/.github/actions/spelling/allow.txt b/.github/actions/spelling/allow.txt
@@ -359,6 +359,7 @@ cer
 cfbundle
 chatbots
 chromadb
+cimg
 claude
 clickable
 cmap
@@ -378,6 +379,7 @@ constexpr
 corpuses
 csa
 cse
+ctd
 cupertino
 dask
 dataframe

diff --git a/gemini/prompts/prompt_optimizer/vapo_lib.py b/gemini/prompts/prompt_optimizer/vapo_lib.py
@@ -21,15 +21,12 @@
 from IPython.core.display import DisplayHandle
 from IPython.display import HTML, display
 from google.cloud import aiplatform, storage
-from google.colab import output
 import ipywidgets as widgets
 import jinja2
 import jinja2.meta
 import pandas as pd
 from tensorflow.io import gfile
 
-output.enable_custom_widget_manager()
-
 
 def is_target_required_metric(eval_metric: str) -> bool:
     """Check if the metric requires the target label."""

diff --git a/gemini/prompts/prompt_optimizer/vertex_ai_prompt_optimizer_ui.ipynb b/gemini/prompts/prompt_optimizer/vertex_ai_prompt_optimizer_ui.ipynb
@@ -291,8 +291,6 @@
         "import json\n",
         "import time\n",
         "\n",
-        "from google.colab import auth\n",
-        "\n",
         "timestamp = datetime.datetime.now().strftime(\"%Y-%m-%dT%H:%M:%S\")\n",
         "display_name = f\"pt_{timestamp}\"\n",
         "\n",