Skip to content

Commit b01655a

Browse files
committed
handle additional dependencies for specific models
1 parent 70db93e commit b01655a

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

scripts/run_model_tests.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,8 @@ case $MODEL_NAME in
119119
print_heading "Error: AWS_REGION environment variable is not set"
120120
exit 1
121121
fi
122+
print_info "Installing anthropic package for AWS-- required in tests"
123+
pip install anthropic
122124
;;
123125
"cohere")
124126
if [ -z "${CO_API_KEY}" ]; then
@@ -143,6 +145,8 @@ case $MODEL_NAME in
143145
print_heading "Error: MISTRAL_API_KEY environment variable is not set"
144146
exit 1
145147
fi
148+
print_info "Installing groq package for Mistral-- required in tests"
149+
pip install groq
146150
;;
147151
"nvidia")
148152
if [ -z "${NVIDIA_API_KEY}" ]; then

0 commit comments

Comments
 (0)