slide_generation/pyproject.toml

[tool.poetry]
name = "slide-generation"
version = "0.1.0"
description = "The public repository of the slide generation project."
authors = ["Yoshiki Kubotani <yoshikikubotani.lab@gmail.com>"]
license = "MIT"
readme = "README.md"

[tool.poetry.dependencies]
python = ">=3.8.1,<3.11"
jaxlib = {version = "^0.3.25+cuda11.cudnn82", source = "jax-cuda"}
jax = "^0.3.25"
flax = "^0.6.2"
hydra-core = "^1.2.0"
wandb = "^0.13.5"
tensorflow = "^2.11.0"
transformers = "^4.25.1"
nltk = "^3.8"
scann = "^1.2.9"

[tool.poetry.group.dev.dependencies]
black = "^22.10.0"
pytest-cov = "^4.0.0"
isort = "^5.10.1"
flake8 = "^6.0.0"
mypy = "^0.991"
mdformat-gfm = "^0.3.5"
mdformat_frontmatter = "^0.4.1"
mdformat_footnote = "^0.1.1"
paddlepaddle-gpu = { url = "https://paddle-wheel.bj.bcebos.com/release/2.4/linux/linux-gpu-cuda11.2-cudnn8-mkl-gcc8.2-avx/paddlepaddle_gpu-2.4.1.post112-cp38-cp38-linux_x86_64.whl" }
ipykernel = "^6.19.2"
paddleocr = "^2.6.1.1"
pdfplumber = "^0.7.6"


[[tool.poetry.source]]
name = "jax-cuda"
url = "https://storage.googleapis.com/jax-releases/jax_cuda_releases.html"
default = false
secondary = false

[tool.black]
line-length = 88

[tool.isort]
profile = "black"
line_length = 88

[tool.mypy]
python_version = 3.8
# following setting is same as pysen
# https://github.com/pfnet/pysen/blob/main/setup.cfg#L12
check_untyped_defs = true
disallow_any_decorated = false
disallow_any_generics = false
disallow_any_unimported = false
disallow_incomplete_defs = true
disallow_subclassing_any = true
disallow_untyped_calls = true
disallow_untyped_decorators = false
disallow_untyped_defs = true
ignore_errors = false
ignore_missing_imports = true
no_implicit_optional = true
show_error_codes = true
strict_equality = true
strict_optional = true
warn_redundant_casts = true
warn_return_any = true
warn_unreachable = true
warn_unused_configs = true
warn_unused_ignores = true

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