pyproject.toml

[tool.poetry]
name = "vital"
version = "0.1.0"
description = "Deep Learning Tooling from the Videos & Images Theory and Analytics Laboratory at UdeS"
authors = ["Nathan Painchaud <nathan.painchaud@usherbrooke.ca>"]
license = "Apache-2.0"
readme = "README.md"
repository = "https://github.com/vitalab/vital"
classifiers=[
    "Environment :: Console",
    "Natural Language :: English",
    # Pick your license as you wish
    "License :: OSI Approved :: Apache Software License",
    "Operating System :: OS Independent",
    # Specify the Python versions you support here. In particular, ensure
    # that you indicate whether you support Python 2, Python 3 or both.
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.10",
]
packages = [
    { include = "vital" },
    { include = "hydra_plugins" },
]

[tool.poetry.dependencies]
python = "~3.10.6"
torch = "~2.0.0"
torchvision = "~0.15.1"
pytorch-lightning = { version = "~2.0.0", extras = ["extra"] }
hydra-core = "~1.3.0"
torchmetrics = "*"
torchinfo = "*"
pathos = "*"
pillow = "*"
numpy = "*"
scipy = "*"
seaborn = "*"
matplotlib = "*"
umap-learn = { version = "*", extras = ["plot"] }
pacmap = "*"
pandas = "*"
h5py = "*"
PyYAML = "*"
StrEnum = "*"
tqdm = "*"
SimpleITK = "*"
medpy = "*"
natsort = "*"
nibabel = "*"
albumentations = "*"
comet-ml = "*"
python-dotenv = "*"
scikit-image = "*"
scikit-learn = "*"
Keras-Preprocessing = "*"

[tool.poetry.group.dev.dependencies]
pre-commit = "^2.20.0"
isort = "==5.12.0"
black = "==23.3.0"
flake8 = "==6.0.0"
flake8-docstrings = "==1.7.0"

[tool.poetry.scripts]
vital-runner = "vital.runner:main"
bmode-pol2cart = "vital.utils.image.us.bmode:main"
camus-hdf5 = "vital.data.camus.dataset_generator:main"
acdc-hdf5 = "vital.data.acdc.dataset_generator:main"
anatomical-da = "vital.data.augmentation.anatomically_constrained_da:main"

[tool.black]
line-length = 120
exclude = "(.eggs|.git|.hg|.mypy_cache|.nox|.tox|.venv|.svn|_build|buck-out|build|dist)"

[tool.isort]
profile = "black"
line_length = 120
src_paths = ["vital"]

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