chore(dependencies): Add Watsonx LLM support to LMEval Job image (#360)

trustyai-explainability · Nov 14, 2024 · 8f3b5c5 · 8f3b5c5
1 parent 40080ed
commit 8f3b5c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.lmes-job b/Dockerfile.lmes-job
@@ -15,7 +15,7 @@ ENV PATH="/opt/app-root/bin:/opt/app-root/src/.local/bin/:/opt/app-root/src/bin:
 # Clone the Git repository, check out v0.4.4 and install the Python package
 RUN git clone https://github.com/opendatahub-io/lm-evaluation-harness.git && \
     cd lm-evaluation-harness &&  git checkout release-0.4.5 && \
-    pip install --no-cache-dir --user -e .[api]
+    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