pyproject.toml

[tool.poetry]
name = "groovis"
version = "0.1.0"
description = ""
authors = ["junhsss <junhsssr@gmail.com>"]

[tool.poetry.dependencies]
python = ">=3.9,<3.11"
Pillow = "^9.0.1"
torch = "=1.12.1"
einops = "^0.6.0"
albumentations = "^1.2.1"
matplotlib = "^3.5.2"
torchvision = "=0.13.1"
wandb = "^0.13.3"
python-dotenv = "^0.21.0"
omegaconf = "^2.2.3"
torchmetrics = "^0.9.3"
loguru = "^0.6.0"
datasets = "^2.5.1"
torch-optimizer = "^0.3.0"
pytorch-lightning = "^1.7.7"
rich = "^12.6.0"
timm = "^0.6.11"
hydra-core = "^1.2.0"
hydra-zen = "v0.9.0rc5"
jaxtyping = "^0.2.9"
beartype = "^0.11.0"

[tool.poetry.dev-dependencies]
black = "^22.1.0"
flake8 = "^4.0.1"
pytest = "^7.0.1"
ImageHash = "^4.2.1"
pytest-watch = "^4.2.0"
pytest-testmon = "^1.3.4"
pre-commit = "^2.20.0"
mypy = "^0.982"


[tool.black]
line-length = 88

[tool.isort]
profile = "black"

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