Pin pip dependency versions for notebook (#291)

Co-authored-by: Umesh Kumhar <[email protected]>
GoogleCloudPlatform · Mar 6, 2024 · 3326570 · 3326570
1 parent 7093da3
commit 3326570
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/applications/rag/example_notebooks/rag-kaggle-ray-sql-latest.ipynb b/applications/rag/example_notebooks/rag-kaggle-ray-sql-latest.ipynb
@@ -7,8 +7,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install langchain ray==2.7.1 datasets sentence-transformers kaggle\n",
-    "!pip install \"cloud-sql-python-connector[pg8000]\" SQLAlchemy==2.0.7"
+    "!pip install langchain==0.1.9 ray==2.7.1 datasets==2.18.0 sentence-transformers==2.5.1 kaggle==1.6.6\n",
+    "!pip install \"cloud-sql-python-connector[pg8000]==1.7.0\" SQLAlchemy==2.0.7"
    ]
   },
   {
@@ -294,14 +294,14 @@
     "    runtime_env={\n",
     "        \"working_dir\": \"/home/jovyan/test\", # upload the local working directory to ray workers\n",
     "        \"pip\": [\n",
-    "                \"langchain\",\n",
+    "                \"langchain==0.1.9\",\n",
     "                \"transformers\",\n",
-    "                \"sentence-transformers\",\n",
+    "                \"sentence-transformers==2.5.1\",\n",
     "                \"pyarrow\",\n",
-    "                \"datasets\",\n",
+    "                \"datasets==2.18.0\",\n",
     "                \"torch==2.0.1\",\n",
-    "                \"cloud-sql-python-connector[pg8000]\",\n",
-    "                \"SQLAlchemy\",\n",
+    "                \"cloud-sql-python-connector[pg8000]==1.7.0\",\n",
+    "                \"SQLAlchemy==2.0.7\",\n",
     "                \"huggingface_hub\",\n",
     "                ]\n",
     "    }\n",