pyproject.toml

[tool.pytest.ini_options]
minversion = "6.2"
addopts = "-ra"
testpaths = ["tests"]

[tool.coverage.report]
exclude_lines = [
  "raise NotImplementedError",
  "raise NotImplementedError()",
  "pragma: nocover",
  "if __name__ == .__main__.:",
]
[tool.mypy]
files = ["src/**/*.py", "test/*.py"]
ignore_missing_imports = true

[tool.pydocstyle]
convention = 'google'
# ignore all missing docs errors
add-ignore = ['D100', 'D101', 'D102', 'D103', 'D104', 'D105', 'D106', 'D107']

[tool.bandit]
skips = ["B101"]

[tool.ruff]
line-length = 120

[tool.ruff.format]
docstring-code-format = true
skip-magic-trailing-comma = true

[tool.ruff.lint]
extend-select = ['I']

[tool.ruff.lint.isort]
known-third-party = ["numpy", "pytest", "wandb", "torch"]
known-first-party = ["nn_core"]
known-local-folder = ["lightningsparseinst"]
split-on-trailing-comma = false

[tool.ruff.lint.pydocstyle]
convention = "google"

[tool.poetry]
name = "lightningsparseinst"
version = "0.1.0"
description = "SparseInst (CVPR 2022) in PyTorch Lightning"
authors = ["Charitarth Chugh <charitarth.chughg@gmail.com>"]
license = "MIT"
readme = "README.md"
package-mode = true

[tool.poetry.dependencies]
python = "^3.11, <3.13"
torch = "^2.5.0"
nn-template-core = { git = "https://github.com/charitarthchugh/nn-template-core" } # use fork of nn-template-core to allow for newer versions of lightning
lightning = "^2.4.0"
hydra-core = "^1.3.2"
wandb = "^0.18.5"
torchmetrics = "^1.5.1"
omegaconf = "^2.3.0"
fiftyone = "^1.0.1"
albumentations = "^1.4.21"
orjson = "^3.10.12"
lmdb = "^1.5.1"
pycocotools = "^2.0.8"
pillow-simd = "^9.5.0.post2"
opencv-contrib-python = "^4.10.0.84"
timm = "^1.0.11"


[tool.poetry.group.dev.dependencies]
ruff = "^0.6.4"
isort = "^5.13.2"
tqdm = "^4.66.5"
cruft = "^2.15.0"
yappi = "^1.6.10"
pyinstrument = "^5.0.0"
ipywidgets = "^8.1.5"

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