run_infer.sh 838 B

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. #!/bin/bash
  2. set -eo pipefail
  3. source "evaluation/utils/version_control.sh"
  4. MODEL_CONFIG=$1
  5. COMMIT_HASH=$2
  6. AGENT=$3
  7. EVAL_LIMIT=$4
  8. DATASET="biocoder"
  9. checkout_eval_branch
  10. if [ -z "$AGENT" ]; then
  11. echo "Agent not specified, use default CodeActAgent"
  12. AGENT="CodeActAgent"
  13. fi
  14. get_agent_version
  15. echo "AGENT: $AGENT"
  16. echo "AGENT_VERSION: $AGENT_VERSION"
  17. echo "MODEL_CONFIG: $MODEL_CONFIG"
  18. echo "DATASET: $DATASET"
  19. COMMAND="poetry run python evaluation/biocoder/run_infer.py \
  20. --agent-cls $AGENT \
  21. --llm-config $MODEL_CONFIG \
  22. --max-iterations 10 \
  23. --max-chars 10000000 \
  24. --eval-num-workers 1 \
  25. --eval-note ${AGENT_VERSION}_${DATASET}"
  26. if [ -n "$EVAL_LIMIT" ]; then
  27. echo "EVAL_LIMIT: $EVAL_LIMIT"
  28. COMMAND="$COMMAND --eval-n-limit $EVAL_LIMIT"
  29. fi
  30. # Run the command
  31. echo $COMMAND
  32. eval $COMMAND
  33. checkout_original_branch