pyproject.toml

[project]
name = "fibad"
license = {file = "LICENSE"}
readme = "README.md"
authors = [
    { name = "LINCC Frameworks", email = "mtauraso@uw.edu" }
]
classifiers = [
    "Development Status :: 4 - Beta",
    "License :: OSI Approved :: MIT License",
    "Intended Audience :: Developers",
    "Intended Audience :: Science/Research",
    "Operating System :: OS Independent",
    "Programming Language :: Python",
]
dynamic = ["version"]
requires-python = ">=3.9"
dependencies = [
    "astropy", # Used to load fits files of sources to query HSC cutout server
    "pytorch-ignite", # Used for distributed training, logging, etc.
    "toml", # Used to load configuration files as dictionaries
    "tomlkit", # Used to load configuration files as dictionaries and retain comments
    "torch", # Used for CNN model and in train.py
    "torchvision", # Used in hsc data loader, example autoencoder, and CNN model data set
    "tensorboardX", # Used to log training metrics
    "tensorboard", # Used to log training metrics
    "GPUtil", # Used to gather GPU usage information
    "schwimmbad", # Used to speedup hsc data loader file scans
    "chromadb", # Used for similarity search
    "holoviews", # Used in Holoviews visualization prototype
    "bokeh", # Used in Holoviews visualization prototype
    "jupyter_bokeh", # Used in Holoviews visualization prototype
    "datashader", # Used in Holoviews visualization prototype
    "pandas", # Used in Holoviews visualization prototype
    "numpy", # Used in Holoviews visualization prototype
    "scipy", # Used in Holoviews visualization prototype
    "cython", # Used in Holoviews visualization prototype
    "mlflow", # Used to log training metrics and compare models
    "pynvml", # Used to gather GPU usage information
    "umap-learn", # Used to map latent spaces down to 2d
    "pooch", # Used to download data files
    "onnx", # Used to export models to ONNX format
    "onnxruntime", # Used to run ONNX models
]

[project.scripts]
fibad = "fibad_cli.main:main"

[project.urls]
"Source Code" = "https://github.com/lincc-frameworks/fibad"

# On a mac, install optional dependencies with `pip install '.[dev]'` (include the single quotes)
[project.optional-dependencies]
examples = [
]

dev = [
    "asv==0.6.4", # Used to compute performance benchmarks
    "jupyter", # Clears output from Jupyter notebooks
    "matplotlib", # For example notebooks
    "mypy", # Used for static type checking of files
    "pre-commit", # Used to run checks before finalizing a git commit
    "pytest",
    "pytest-cov", # Used to report total code coverage
    "ruff", # Used for static linting of files
]

[build-system]
requires = [
    "setuptools>=62", # Used to build and package the Python project
    "setuptools_scm>=6.2", # Gets release version from git. Makes it available programmatically
]
build-backend = "setuptools.build_meta"

[tool.setuptools_scm]
write_to = "src/fibad/_version.py"

[tool.pytest.ini_options]
testpaths = [
    "tests",
]

[tool.black]
line-length = 110
target-version = ["py39"]

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

[tool.ruff]
line-length = 110
target-version = "py39"

[tool.ruff.lint]
select = [
    # pycodestyle
    "E",
    "W",
    # Pyflakes
    "F",
    # pep8-naming
    "N",
    # pyupgrade
    "UP",
    # flake8-bugbear
    "B",
    # flake8-simplify
    "SIM",
    # isort
    "I",
    # docstrings
    "D101",
    "D102",
    "D103",
    "D106",
    "D206",
    "D207",
    "D208",
    "D300",
    "D417",
    "D419",
    # Numpy v2.0 compatibility
    "NPY201",
]

ignore = [
    "UP006", # Allow non standard library generics in type hints
    "UP007", # Allow Union in type hints
    "SIM114", # Allow if with same arms
    "B028", # Allow default warning level
    "SIM117", # Allow nested with
    "UP015", # Allow redundant open parameters
    "UP028", # Allow yield in for loop
]
[tool.setuptools.package-data]
fibad = ["py.typed"]

[tool.coverage.run]
omit=["src/fibad/_version.py"]

[tool.mypy]
warn_return_any = true
warn_unused_configs = true

[[tool.mypy.overrides]]
module = [
    'torchvision',
    'astropy',
    'astropy.io',
    'schwimmbad',
    'torchvision.transforms',
    'torchvision.datasets',
    'torchvision.transforms.v2',
    'tensorboardX',
    'astropy.table',
    'GPUtil',
    'toml',
    'umap',
    ]
ignore_missing_imports = true