From 41fde7ee6d7c44be042feecc616731768429a4ce Mon Sep 17 00:00:00 2001 From: Anu Reddy Date: Tue, 5 Mar 2024 19:03:45 -0800 Subject: [PATCH] Pin pip dependency versions for notebook --- .../rag-kaggle-ray-sql-latest.ipynb | 14 +++++++------- 1 file changed, 7 insertions(+), 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 index 849eaec20..72d3d5915 100644 --- 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",