pyproject.toml

[tool.poetry]
name = "omlmd"
version = "0.1.6"
description = "OCI Artifact for ML model & metadata"
authors = ["Matteo Mortari <matteo.mortari@gmail.com>"]
readme = "README.md"

[tool.poetry.urls]
Homepage = "https://containers.github.io/omlmd"
Documentation = "https://containers.github.io/omlmd"
Repository = "https://github.com/containers/omlmd"
Issues = "https://github.com/containers/omlmd/issues"
Changelog = "https://github.com/containers/omlmd/releases"

[tool.poetry.dependencies]
python = "^3.9"
oras = "^0.2.23"
pyyaml = "^6.0.1"
click = "^8.1.7"
cloup = "^3.0.5"

[tool.poetry.group.dev.dependencies]
pytest = "^8.3.2"
pytest-mock = "^3.14.0"
jq = "^1.7.0"
scikit-learn = "^1.5.0"
ipykernel = "^6.29.4"
nbconvert = "^7.16.4"
markdown-it-py = "^3.0.0"
model-registry = ">=0.2.9,<0.3.0"
ruff = "^0.6.1"
mypy = "^1.11.1"
types-pyyaml = "^6.0.12.20240808"
types-requests = "^2.32.0.20241016"

[tool.poetry.scripts]
omlmd = "omlmd.cli:cli"

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

[tool.pytest.ini_options]
markers = [
    "e2e: end-to-end testing with localhost:5001",
    "e2e_model_registry: end-to-end testing with localhost:5001 and Kubeflow Model Registry",
]

[tool.ruff]
target-version = "py39"
respect-gitignore = true

[tool.ruff.lint.per-file-ignores]
"*.ipynb" = [
    "E402",
] # exclude https://docs.astral.sh/ruff/rules/module-import-not-at-top-of-file/#notebook-behavior from linting, especially for demos.

[tool.mypy]
python_version = "3.9"
strict = false
pretty = true
show_column_numbers = true
show_error_codes = true
show_error_context = true
ignore_missing_imports = true