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

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

[requires]
python_version = "3.10"

[packages]
torch = {index = "pytorch", version = "==2.2.0+cu118"}
tqdm = ">=4.65.0, <5.0.0"
transformers = "==4.40.1"
numpy = ">=1.26.0, <2.0.0"
pandas = ">=2.2.0, <3.0.0"
scikit-learn = ">=1.4.0, <2.0.0"
boto3 = ">=1.20.24, <2.0.0"
SQLAlchemy = ">=2.0.25, <3.0.0"
dill = ">=0.3.4, <0.4.0"
pyarrow = ">=14.0.1"
kaggle = ">=1.5.12, <2.0.0"
coolname = ">=2.2.0, <3.0.0"
bokeh = ">=3.2.1, <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.21.1"
bitsandbytes = "==0.42.0"
accelerate = "==0.27.2"
openai = ">=1.12.0, <2.0.0"
einops = "==0.7.0"
datasets = ">=2.11.0, <3.0.0"
neptune = "==1.9.1"
Jinja2 = ">=3.1.3, <4.0.0"
h2o-wave = "==1.1.2"
tiktoken = "==0.6.0"
hf-transfer = "==0.1.5"
peft = "==0.9.0"
azure-storage-file-datalake = ">=12.12.0"
deepspeed = "==0.13.2"
keyring = "==24.3.1"

[dev-packages]
black = "==24.3.0"
coverage = "==7.4.3"
flake8 = "==7.0.0"
flake8-black = "==0.3.6"
flake8-isort = "==6.1.1"
isort = "==5.13.2"
mypy = "==1.8.0"
pytest = "==8.0.0"  # >=8.0.1 is not supported by transformers https://github.com/huggingface/transformers/issues/29155
pytest-cov = "==4.1.0"
pytest-dependency = "==0.6.0"
pytest-html = "4.1.1"
types-pyyaml = ">=6.0"
types-requests = ">=2.31"
types-toml = ">=0.10"
wheel = "==0.42.0"
pytest-bdd = "==7.0.1"
hac-playwright = { file = "http://h2o-public-test-data.s3.amazonaws.com/e2e-testing/hac_playwright-1.38.0-py3-none-any.whl" }
pytest-base-url = "==2.1.0"