pyproject.toml

[build-system]
requires = ["hatchling", "versioningit"]
build-backend = "hatchling.build"

[project]
name = "madengine"
dynamic = ["version"]
authors = [
  { name="Advanced Micro Devices", email="mad.support@amd.com" },
]
description = "MAD Engine is a set of interfaces to run various AI models from public MAD."
readme = "README.md"
requires-python = ">=3.8"
dependencies = [
  "pandas",
  "GitPython",
  "jsondiff",
  "sqlalchemy",
  "paramiko",
  "tqdm",
  "typing-extensions",
  "pymongo",
  "toml",
  "typer>=0.9.0",
  "rich>=13.0.0",
  "click>=8.0.0",
  "jinja2>=3.0.0",
  "pyyaml>=6.0",
  "hydra-core>=1.3",
  "omegaconf>=2.3",
  "kubernetes>=28.0.0",
  "pytest>=7.0",
  "pytest-cov",
  "pytest-xdist",
  "pytest-timeout",
  "pytest-mock",
  "pytest-asyncio",
  "black>=21.0.0",
  "flake8",
  "mypy>=0.910",
  "isort",
  "pre-commit",
]
classifiers = [
  "Programming Language :: Python :: 3",
  "License :: OSI Approved :: MIT License",
  "Operating System :: OS Independent",
]

[project.optional-dependencies]
# Host-side trace analysis for `madengine report tracelens`. Kept optional
# because TraceLens pins protobuf>=6.31 and xprof, which can conflict with other
# packages in a shared environment.
tracelens = [
  "TraceLens @ git+https://github.com/AMD-AGI/TraceLens.git@6f9bcdbf6cc9911eb650de57b345917ea4d31a17",
]

[project.scripts]
madengine = "madengine.cli.app:cli_main"

[project.urls]
Homepage = "https://github.com/ROCm/madengine"
Issues = "https://github.com/ROCm/madengine/issues"


[tool.hatch.metadata]
# The tracelens extra installs TraceLens from git; hatchling rejects such direct
# references during metadata generation unless they are allowed here.
allow-direct-references = true

[tool.hatch.build.targets.wheel]
# scripts/ is listed in .gitignore (to exclude external MAD project scripts/ dirs during dev).
# artifacts bypasses VCS exclusion without risk of duplicate-file errors from force-include.
# configs/ Hydra groups must ship in the wheel for --config compose.
artifacts = [
  "src/madengine/scripts/**",
  "src/madengine/configs/**",
]

[tool.hatch.version]
source = "versioningit"

[tool.versioningit.vcs]
method = "git"
default-tag = "v2.0.0"

[tool.versioningit.tag2version]
regex = "v(?P<version>.*)"

[tool.versioningit.format]
distance = "{base_version}.post{distance}+{vcs}{rev}"
dirty = "{base_version}+d{build_date:%Y%m%d}"
distance-dirty = "{base_version}.post{distance}+{vcs}{rev}.d{build_date:%Y%m%d}"

# Code formatting and linting configuration
[tool.black]
line-length = 88
target-version = ['py38', 'py39', 'py310', 'py311']
include = '\.pyi?$'
extend-exclude = '''
/(
  # directories
  \.eggs
  | \.git
  | \.hg
  | \.mypy_cache
  | \.tox
  | \.venv
  | build
  | dist
)/
'''

[tool.isort]
profile = "black"
multi_line_output = 3
line_length = 88
known_first_party = ["madengine"]
known_third_party = ["pytest", "pandas", "numpy", "sqlalchemy"]

[tool.mypy]
python_version = "3.8"
warn_return_any = true
warn_unused_configs = true
disallow_untyped_defs = false
disallow_incomplete_defs = false
check_untyped_defs = true
disallow_untyped_decorators = false
no_implicit_optional = true
warn_redundant_casts = true
warn_unused_ignores = true
warn_no_return = true
warn_unreachable = true
strict_equality = true

[[tool.mypy.overrides]]
module = [
    "paramiko.*",
    "pymongo.*",
    "toml.*",
    "jsondiff.*",
    "git.*",
    "kubernetes.*",
    "yaml.*",
]
ignore_missing_imports = true

[tool.pytest.ini_options]
testpaths = ["tests/unit", "tests/integration", "tests/e2e"]
pythonpath = ["src"]
addopts = "-v --tb=short -ra --strict-markers -W default"
minversion = "7.0"
filterwarnings = [
    "ignore::DeprecationWarning",
    "ignore::PendingDeprecationWarning",
]
markers = [
    "unit: Fast unit tests (no external dependencies)",
    "integration: Integration tests (may be slower, test multiple components)",
    "e2e: End-to-end tests (require full environment, Docker, may be very slow)",
    "slow: Slow tests (can be skipped with -m \"not slow\")",
    "gpu: Tests that require GPU hardware",
    "amd: Tests specific to AMD GPUs",
    "nvidia: Tests specific to NVIDIA GPUs",
    "cpu: Tests for CPU-only execution",
    "requires_docker: Tests that require Docker daemon",
    "requires_models: Tests that require model fixtures",
]

[tool.coverage.run]
source = ["src/madengine"]
omit = [
    "*/tests/*",
    "*/test_*",
    "*/__pycache__/*",
]

[tool.coverage.report]
exclude_lines = [
    "pragma: no cover",
    "def __repr__",
    "if self.debug:",
    "if settings.DEBUG",
    "raise AssertionError",
    "raise NotImplementedError",
    "if 0:",
    "if __name__ == .__main__.:",
    "class .*\\bProtocol\\):",
    "@(abc\\.)?abstractmethod",
]

[tool.bandit]
exclude_dirs = ["tests", "venv", "build", "dist", ".git"]
skips = ["B101"]  # Skip assert_used (used in tests)