pyproject.toml

[project]
name = "snax-mlir"
description = "Driving SNAX with MLIR"
version = "0.2.2"

dependencies = [
    "xdsl @ git+https://github.com/xdslproject/xdsl.git@0d7eae3a1c9ee128986d6729ffc1d49786f7819d"
]

[project.optional-dependencies]

# All optional dependencies for plotting and parsing
dev = [
    "pre-commit",
    "filecheck==1.0.0",
    "lit",
    "numpy<2.0.0",
    "pytest",
]

# All optional dependencies for graphical plotting
viz = [
    "pandas",
    "matplotlib",
]

# Dependencies for neural network compilation
nn = [
    "tensorflow-cpu==2.15.0",
]

[project.scripts]
snax-opt = "compiler.tools.snax_opt_main:main"

[tool.ruff]
select = ["E", "F", "W", "I", "UP", "PT", "TID251"]
ignore = [
    "E741",  # https://beta.ruff.rs/docs/rules/ambiguous-variable-name/
    "PT006", # https://beta.ruff.rs/docs/rules/pytest-parametrize-names-wrong-type/
    "PT007", # https://beta.ruff.rs/docs/rules/pytest-parametrize-values-wrong-type/
    "PT011", # https://beta.ruff.rs/docs/rules/pytest-raises-too-broad/
    "PT012", # https://beta.ruff.rs/docs/rules/pytest-raises-with-multiple-statements/
    "PT015", # https://beta.ruff.rs/docs/rules/pytest-assert-always-false/
]
target-version = "py310"
line-length = 120

[tool.ruff.flake8-tidy-imports.banned-api]
"xdsl.parser.core".msg = "Use xdsl.parser instead."
"xdsl.parser.attribute_parser".msg = "Use xdsl.parser instead."
"xdsl.parser.affine_parser".msg = "Use xdsl.parser instead."
"xdsl.ir.core".msg = "Use xdsl.ir instead."
"xdsl.irdl.irdl".msg = "Use xdsl.irdl instead"
"xdsl.ir.affine.affine_expr".msg = "Use xdsl.ir.affine instead"
"xdsl.ir.affine.affine_map".msg = "Use xdsl.ir.affine instead"
"xdsl.ir.affine.affine_set".msg = "Use xdsl.ir.affine instead"


[tool.ruff.per-file-ignores]
"__init__.py" = ["F403"]
"tests/filecheck/frontend/programs/invalid.py" = ["F811", "F841"]
"tests/filecheck/frontend/dialects/invalid.py" = ["F811"]
"tests/test_declarative_assembly_format.py" = ["F811"]
"versioneer.py" = ["ALL"]
"_version.py" = ["ALL"]

[tool.ruff.mccabe]
max-complexity = 10