[[source]]
name = "pypi"
url = "https://pypi.org/simple"
verify_ssl = true

[[source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu121"
verify_ssl = false

[requires]
python_version = "3.10"

[packages]
torch = {index = "pytorch", version = "==2.4.0+cu121"}
tqdm = ">=4.65.0, <5.0.0"
transformers = "==4.45.0"
numpy = ">=1.26.0, <2.0.0"
pandas = ">=2.2.0, <3.0.0"
scikit-learn = ">=1.5.1, <2.0.0"
boto3 = ">=1.20.24, <2.0.0"
SQLAlchemy = ">=2.0.25, <3.0.0"
dill = ">=0.3.8, <0.4.0"
pyarrow = ">=14.0.1"
kaggle = ">=1.6.14, <2.0.0, !=1.6.15"
coolname = ">=2.2.0, <3.0.0"
bokeh = ">=3.5.0, <4.0.0"
beautifulsoup4 = ">=4.11.1, <5.0.0"
sqlitedict = "==1.7.0"
sentencepiece = ">=0.1.96, <0.2.0"
sacrebleu = "==2.0.0"
toml = ">=0.10.2, <0.11.0"
pyyaml = ">=6.0.0, <7.0.0"
protobuf = "==3.20.3"
fastparquet = ">=2023.7.0"
gputil = ">=1.4.0, <2.0.0"
huggingface-hub = "==0.24.6"
bitsandbytes = "==0.43.3"
accelerate = "==0.34.2"
openai = ">=1.12.0, <2.0.0"
einops = "==0.8.0"
datasets = ">=2.20.0, <3.0.0"
neptune = "==1.10.4, <2.0.0"
wandb = ">=0.17.5, <1.0.0"
Jinja2 = ">=3.1.3, <4.0.0"
h2o-drive = ">=3.0.0, <4.0.0"
h2o-wave = ">=1.4.0, <2.0.0"
tiktoken = "==0.7.0"
hf-transfer = "==0.1.8"
peft = "==0.12.0"
azure-storage-file-datalake = ">=12.12.0"
deepspeed = "==0.14.4"
keyring = ">=25.2.1, <26.0.0"
pydantic = ">=2.8.2, <3.0.0"

[dev-packages]
black = "==24.4.2"
coverage = "==7.5.1"
flake8 = "==7.0.0"
flake8-black = "==0.3.6"
flake8-isort = "==6.1.1"
isort = "==5.13.2"
mypy = "==1.10.0"
pytest = "==8.2.0"
pytest-cov = "==5.0.0"
pytest-dependency = "==0.6.0"
pytest-html = ">=4.1.1, <5.0.0"
types-pyyaml = ">=6.0"
types-requests = ">=2.31"
types-toml = ">=0.10"
wheel = "==0.43.0"
pytest-bdd = "==7.1.2"
hac-playwright = { file = "http://h2o-public-test-data.s3.amazonaws.com/e2e-testing/hac_playwright-1.45.0-py3-none-any.whl" }
pytest-base-url = "==2.1.0"
pytest-mock = "==3.14.0"