datasets<4.0,>=2.16.0
evaluate
scipy>=1.10.1
diskcache

[all]
unitxt[base]
unitxt[dev]
unitxt[docs]
unitxt[helm]
unitxt[service]
unitxt[tests]
unitxt[ui]
unitxt[watsonx]
unitxt[assistant]
unitxt[text2sql]

[assistant]
streamlit
watchdog
litellm

[bluebench]
unitxt[remote_inference]
unitxt[local_inference]
conllu
scikit-learn
sympy
bert_score
nltk
rouge_score
sacrebleu[ko]

[dev]
ruff
pre-commit
detect-secrets
tomli
codespell
fuzzywuzzy
httpretty
psutil

[docs]
sphinx_rtd_theme
piccolo_theme
sphinxext-opengraph
datasets<4.0,>=2.16.0
evaluate
nltk
rouge_score
scikit-learn
jiwer
editdistance
fuzzywuzzy
pydantic

[helm]
crfm-helm[unitxt]>=0.5.3

[inference-tests]
litellm>=1.52.9
tenacity
diskcache
numpy==1.26.4
ollama

[local_inference]
transformers
torch
accelerate

[remote_inference]
litellm>=1.52.9
tenacity
diskcache

[service]
torch==1.12.1
fastapi==0.109.0
uvicorn[standard]==0.27.0.post1
python-jose[cryptography]==3.3.0
transformers

[tests]
bert_score
transformers
sentence_transformers
ibm-cos-sdk
kaggle==1.6.14
opendatasets
httpretty~=1.1.4
editdistance
rouge-score
nltk
sacrebleu[ja,ko]
scikit-learn<=1.5.2
jiwer
conllu
llama-index-core
llama-index-llms-openai
pytrec-eval
SentencePiece
fuzzywuzzy
openai
ibm-generative-ai
bs4
tenacity==8.3.0
accelerate
func_timeout==4.3.5
Wikipedia-API
sqlglot
sqlparse
diskcache
pydantic
jsonschema_rs

[text2sql]
sqlglot
func_timeout==4.3.5
sqlparse
tabulate

[ui]
gradio
transformers

[watsonx]
ibm-watsonx-ai==1.2.10
