pyproject.toml

[project]
name = "edge_tta"
version = "0.0.1"
description = "A system level benchmark of online test-time adaptation methods from the literature on resource-constrained hardware."
readme = "README.md"
requires-python = ">=3.10,<3.13"
license = {text = "Apache-2.0"}
authors = [
    {name = "Mateusz Piechocki", email = "mateusz.piechocki@put.poznan.pl"},
]
keywords = ["python", "test-time adaptation", "online learning", "resource-constrained hardware"]
classifiers = [
    "License :: OSI Approved :: Apache Software License",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
]
dependencies = [
    "click",
    "numpy",
    "opencv-contrib-python",
    "pandas",
    "pvlib",
    "tqdm",
    "scikit-learn",
    "psutil",  # For performance monitoring (CPU/memory tracking)
]

[project.optional-dependencies]
test = [
    "pytest",
    "ruff",
    "pre-commit",
]
dev = [
    "edge_tta[test]",
    # --------- pytorch --------- #
    "torch==2.8.0",
    "torchvision==0.23.0",
    "lightning==2.5.3",
    "torchmetrics==1.8.1",
    "timm==1.0.19",
    "torchinfo==1.8.0",
    # --------- loggers --------- #
    "neptune",
    # --------- dvc --------- #
    "dvc",
    # --------- performance monitoring --------- #
    "emt",
    "nvidia-ml-py",
    # --------- misc --------- #
    "cma",
    "python-dotenv",
    "matplotlib",
    "ptflops",
    "pyRAPL",
    "nvidia-ml-py",
    "pymongo",
    # --------- Discrete Wavelet Transform for L-TTA --------- #
    "PyWavelets",
    "pytorch-wavelets",
]

[project.urls]
Repository = "https://github.com/MatPiech/edge-tta"

[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[tool.setuptools.packages.find]
where = ["."]
include = ["edge_tta*"]
exclude = ["tests*", "scripts*", "notebooks*"]

# Pytest configuration
[tool.pytest.ini_options]
minversion = "7.0"
addopts = [
    "--strict-markers",
    "--strict-config",
    "--cov=src",
    "--cov-report=term-missing",
    "--cov-report=html",
    "--cov-report=xml",
]
testpaths = ["tests"]
python_files = ["test_*.py", "*_test.py"]
python_classes = ["Test*"]
python_functions = ["test_*"]
markers = [
    "slow: marks tests as slow (deselect with '-m \"not slow\"')",
    "integration: marks tests as integration tests",
    "unit: marks tests as unit tests",
]

# Ruff configuration
[tool.ruff]
target-version = "py312"
line-length = 127
indent-width = 4

exclude = [
    "notebooks/*",
]

[tool.ruff.lint]
# Enable Pyflakes (`F`) and a subset of the pycodestyle (`E`) codes by default.
select = [
    "E4",     # pycodestyle errors
    "E7",
    "E9",
    "F",      # Pyflakes
    "W",      # pycodestyle warnings
    "I",      # isort
    "N",      # pep8-naming
    "UP",     # pyupgrade
    "B",      # flake8-bugbear
    "A",      # flake8-builtins
    "C4",     # flake8-comprehensions
    "T20",    # flake8-print
    "SIM",    # flake8-simplify
    "ARG",    # flake8-unused-arguments
    "PTH",    # flake8-use-pathlib
    "RUF",    # Ruff-specific rules
]
ignore = [
    "E501",   # Line too long (handled by formatter)
    "B008",   # Do not perform function calls in argument defaults
    "B904",   # Allow `raise` without `from` inside `except`
    "ARG002", # Allow unused arguments in function definitions
    "T201",   # Allow print statements
    "N812",   # Lowercase imported as non-lowercase
    "C419",   # Allow unnecessary list comprehension
]

# Allow fix for all enabled rules (when `--fix`) is provided.
fixable = ["ALL"]
unfixable = []

[tool.ruff.format]
# Like Black, use double quotes for strings.
quote-style = "double"

# Like Black, indent with spaces, rather than tabs.
indent-style = "space"

# Like Black, respect magic trailing commas.
skip-magic-trailing-comma = false

# Like Black, automatically detect the appropriate line ending.
line-ending = "auto"

[tool.ruff.lint.per-file-ignores]
"tests/**/*" = ["T20", "ARG", "SIM"]  # Allow prints and unused args in tests

[tool.ruff.lint.isort]
known-first-party = ["edge_tta"]
force-sort-within-sections = true