pyproject.toml

[tool.poetry]
name = "docmae"
version = "0.1"
description = "Unofficial implementation of DocMAE paper"
authors = ["David Komorowicz <dawars00@gmail.com>"]
packages = [
    { include = "docmae" }  # dir name containing code if differs from name
]

[tool.poetry.dependencies]
python = "~3.10"
opencv-python-headless = "^4.5"
Pillow-SIMD = "^9.0"
torch = [
    {markers = "sys_platform == 'linux'", url="https://download.pytorch.org/whl/cu118/torch-2.0.1%2Bcu118-cp310-cp310-linux_x86_64.whl"},
    {markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", url="https://download.pytorch.org/whl/cpu/torch-2.0.1-cp310-none-macosx_10_9_x86_64.whl"},
    {markers = "sys_platform == 'darwin' and platform_machine == 'arm64'", url="https://download.pytorch.org/whl/cpu/torch-2.0.1-cp310-none-macosx_11_0_arm64.whl"},
    {markers = "sys_platform == 'win32'", url="https://download.pytorch.org/whl/cu118/torch-2.0.1%2Bcu118-cp310-cp310-win_amd64.whl"},
]
torchvision = [
    {markers = "sys_platform == 'linux'", url="https://download.pytorch.org/whl/cu118/torchvision-0.15.2%2Bcu118-cp310-cp310-linux_x86_64.whl"},
    {markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", url="https://download.pytorch.org/whl/cpu/torchvision-0.15.2-cp310-cp310-macosx_10_9_x86_64.whl"},
    {markers = "sys_platform == 'darwin' and platform_machine == 'arm64'", url="https://download.pytorch.org/whl/cpu/torchvision-0.15.2-cp310-cp310-macosx_10_9_x86_64.whl"},
    {markers = "sys_platform == 'win32'", url="https://download.pytorch.org/whl/cu118/torchvision-0.15.2%2Bcu118-cp310-cp310-win_amd64.whl"},
]
lightning = "^2.0.4"
pydantic = "<2.0.0"  # https://github.com/Lightning-AI/lightning/issues/18027
torchmetrics = "^0.11.1"
torchsummary = "^1.5.1"
datasets = "^2.13"
transformers = "^4.30"
accelerate = "^0.20.1"

matplotlib = "^3.6"
numpy = "^1.24"
scikit-image = "^0.19"
scikit-learn = ">=1.0"
tensorboard = "^2.3"
minio = "^7.1.15"
h5py = "^3.9.0"
kornia = "^0.7.0"
gin-config = "^0.5.0"

[tool.poetry.group.test.dependencies]
pytest = "^7.4.0"

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

# optional scripts
[tool.poetry.scripts]
docmae-train = "docmae.train:main"
docmae-pretrain = "docmae.pretrain_pl:main"
docmae-finetune = "docmae.fine_tune:main"

[tool.black]
line-length = 128
include = '\.pyi?$'
exclude = '''
(
  /(
      \.eggs         # exclude a few common directories in the
    | \.git          # root of the project
    | \.hg
    | \.mypy_cache
    | \.pytest_cache
    | \.tox
    | \.venv
    | venv
    | _build
    | buck-out
    | build
    | dist
  )/
)
'''