pyproject.toml

[tool.poetry]
name = "textbook"
version = "0.1.0"
description = ""
authors = ["Jina ai"]
license = "apache 2.0"
readme = "README.md"

[tool.poetry.dependencies]
python = "^3.9"
torch = ">=2.0.1"
transformers = {version = ">=4.30.2", extras = ["deepspeed"]}
typer = {version = "^0.9.0", extras = ["all"]}
wandb = "^0.15.3"
datasets = "^2.13.1"
accelerate = "^0.21.0"
sentencepiece = "^0.1.99"
einops = "^0.6.1"
openai = "^0.27.8"
pydantic = "<2.0.0"
human-eval = {path = "human-eval", develop = true}


[tool.poetry.group.dev.dependencies]
pytest = ">=7.3.1"
black = ">=23.3.0"
isort = ">=5.12.0"
ruff = ">=0.0.269"
pre-commit = ">=3.3.2"
jupyterlab = ">=4.0.0"
jupyterlab-code-formatter = ">=2.2.1"
mypy = ">=1"
pytest-mock = ">=3.11.1"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"

[tool.ruff]
ignore = ["F722"]
line-length = 120

[tool.mypy]
ignore_missing_imports = true

[tool.pytest.ini_options]
markers = [
    "slow: marks tests as slow (deselect with '-m \"not slow\"')",
    "openai: need openai key ",
]