Merge remote-tracking branch 'upstream/main' into rhoai-2.16

red-hat-data-services · Nov 15, 2024 · 29cbe0a · 29cbe0a
2 parents 6932691 + dcce372
commit 29cbe0a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/Dockerfile.lmes-job b/Dockerfile.lmes-job
@@ -16,7 +16,7 @@ RUN mkdir -p /opt/app-root/src/my_catalogs/cards && chmod -R g+rwx /opt/app-root
 RUN mkdir -p /opt/app-root/src/.cache
 ENV PATH="/opt/app-root/bin:/opt/app-root/src/.local/bin/:/opt/app-root/src/bin:/usr/local/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin"
 
-RUN pip install --no-cache-dir --user -e .[api]
+RUN pip install --no-cache-dir --user -e .[api,ibm_watsonx_ai]
 
 RUN python -c 'from lm_eval.tasks.unitxt import task; import os.path; print("class: !function " + task.__file__.replace("task.py", "task.Unitxt"))' > ./my_tasks/unitxt
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -74,6 +74,7 @@ testing = ["pytest==8.3.3", "pytest-cov==6.0.0", "pytest-xdist==3.6.1"]
 vllm = ["vllm>=0.4.2"]
 zeno = ["pandas==2.2.3", "zeno-client"]
 wandb = ["wandb>=0.16.3", "pandas==2.2.3", "numpy==2.1.2"]
+ibm_watsonx_ai = ["ibm-watsonx-ai==1.1.23" ]
 all = [
     "lm_eval[anthropic]",
     "lm_eval[dev]",