mirror of
https://github.com/codeflash-ai/codeflash.git
synced 2026-05-04 18:25:17 +00:00
9f4776eb2e
Replace pre-commit with prek (faster Rust-based alternative) for linting. - Add prek to dev dependencies - Replace pre-commit workflow with prek workflow using setup-uv@v6 - Update Claude workflow allowed tools to use prek
355 lines
7.6 KiB
TOML
355 lines
7.6 KiB
TOML
[project]
|
|
name = "codeflash"
|
|
dynamic = ["version"]
|
|
description = "Client for codeflash.ai - automatic code performance optimization, powered by AI"
|
|
authors = [{ name = "CodeFlash Inc.", email = "contact@codeflash.ai" }]
|
|
requires-python = ">=3.9"
|
|
readme = "README.md"
|
|
license = {text = "BSL-1.1"}
|
|
keywords = [
|
|
"codeflash",
|
|
"performance",
|
|
"optimization",
|
|
"ai",
|
|
"code",
|
|
"machine learning",
|
|
"LLM",
|
|
]
|
|
dependencies = [
|
|
"unidiff>=0.7.4",
|
|
"pytest>=7.0.0",
|
|
"gitpython>=3.1.31",
|
|
"libcst>=1.0.1",
|
|
"jedi>=0.19.1",
|
|
# Tree-sitter for multi-language support
|
|
"tree-sitter>=0.23.0",
|
|
"tree-sitter-javascript>=0.23.0",
|
|
"tree-sitter-typescript>=0.23.0",
|
|
"pytest-timeout>=2.1.0",
|
|
"tomlkit>=0.11.7",
|
|
"junitparser>=3.1.0",
|
|
"pydantic>=1.10.1",
|
|
"humanize>=4.0.0",
|
|
"posthog>=3.0.0",
|
|
"click>=8.1.0",
|
|
"inquirer>=3.0.0",
|
|
"sentry-sdk>=1.40.6,<3.0.0",
|
|
"parameterized>=0.9.0",
|
|
"isort>=5.11.0",
|
|
"dill>=0.3.8",
|
|
"rich>=13.8.1",
|
|
"lxml>=5.3.0",
|
|
"crosshair-tool>=0.0.78",
|
|
"coverage>=7.6.4",
|
|
"line_profiler>=4.2.0",
|
|
"platformdirs>=4.3.7",
|
|
"pygls>=2.0.0,<3.0.0",
|
|
"codeflash-benchmark",
|
|
"filelock",
|
|
"pytest-asyncio>=1.2.0",
|
|
]
|
|
|
|
[project.urls]
|
|
Homepage = "https://codeflash.ai"
|
|
|
|
[project.scripts]
|
|
codeflash = "codeflash.main:main"
|
|
|
|
[project.optional-dependencies]
|
|
|
|
[dependency-groups]
|
|
dev = [
|
|
"ipython>=8.12.0",
|
|
"mypy>=1.13",
|
|
"ruff>=0.7.0",
|
|
"lxml-stubs>=0.5.1",
|
|
"pandas-stubs>=2.2.2.240807, <2.2.3.241009",
|
|
"types-Pygments>=2.18.0.20240506",
|
|
"types-colorama>=0.4.15.20240311",
|
|
"types-decorator>=5.1.8.20240310",
|
|
"types-jsonschema>=4.23.0.20240813",
|
|
"types-requests>=2.32.0.20241016",
|
|
"types-six>=1.16.21.20241009",
|
|
"types-cffi>=1.16.0.20240331",
|
|
"types-openpyxl>=3.1.5.20241020",
|
|
"types-regex>=2024.9.11.20240912",
|
|
"types-python-dateutil>=2.9.0.20241003",
|
|
"types-gevent>=24.11.0.20241230,<25",
|
|
"types-greenlet>=3.1.0.20241221,<4",
|
|
"types-pexpect>=4.9.0.20241208,<5",
|
|
"types-unidiff>=0.7.0.20240505,<0.8",
|
|
"prek>=0.2.25",
|
|
"ty>=0.0.14",
|
|
"uv>=0.9.29",
|
|
]
|
|
tests = [
|
|
"black>=25.9.0",
|
|
"jax>=0.4.30",
|
|
"numpy>=2.0.2",
|
|
"pandas>=2.3.3",
|
|
"pyrsistent>=0.20.0",
|
|
"scipy>=1.13.1",
|
|
"torch>=2.8.0",
|
|
"xarray>=2024.7.0",
|
|
"eval_type_backport",
|
|
"numba>=0.60.0",
|
|
"tensorflow>=2.20.0",
|
|
]
|
|
|
|
[tool.hatch.build.targets.sdist]
|
|
include = ["codeflash"]
|
|
exclude = [
|
|
"docs/*",
|
|
"experiments/*",
|
|
"tests/*",
|
|
"*.pyc",
|
|
"__pycache__",
|
|
"*.pyo",
|
|
"*.pyd",
|
|
"*.so",
|
|
"*.dylib",
|
|
"*.dll",
|
|
"*.exe",
|
|
"*.log",
|
|
"*.tmp",
|
|
".env",
|
|
".env.*",
|
|
"**/.env",
|
|
"**/.env.*",
|
|
".env.example",
|
|
"*.pem",
|
|
"*.key",
|
|
"secrets.*",
|
|
"config.yaml",
|
|
"config.json",
|
|
".git",
|
|
".gitignore",
|
|
".gitattributes",
|
|
".github",
|
|
"Dockerfile",
|
|
"docker-compose.yml",
|
|
"*.md",
|
|
"*.txt",
|
|
"*.csv",
|
|
"*.db",
|
|
"*.sqlite3",
|
|
"*.pdf",
|
|
"*.docx",
|
|
"*.xlsx",
|
|
"*.pptx",
|
|
"*.iml",
|
|
".idea",
|
|
".vscode",
|
|
".DS_Store",
|
|
"Thumbs.db",
|
|
"venv",
|
|
"env",
|
|
]
|
|
|
|
[tool.hatch.build.targets.wheel]
|
|
exclude = [
|
|
"docs/*",
|
|
"experiments/*",
|
|
"tests/*",
|
|
"*.pyc",
|
|
"__pycache__",
|
|
"*.pyo",
|
|
"*.pyd",
|
|
"*.so",
|
|
"*.dylib",
|
|
"*.dll",
|
|
"*.exe",
|
|
"*.log",
|
|
"*.tmp",
|
|
".env",
|
|
".env.*",
|
|
"**/.env",
|
|
"**/.env.*",
|
|
".env.example",
|
|
"*.pem",
|
|
"*.key",
|
|
"secrets.*",
|
|
"config.yaml",
|
|
"config.json",
|
|
".git",
|
|
".gitignore",
|
|
".gitattributes",
|
|
".github",
|
|
"Dockerfile",
|
|
"docker-compose.yml",
|
|
"*.md",
|
|
"*.txt",
|
|
"*.csv",
|
|
"*.db",
|
|
"*.sqlite3",
|
|
"*.pdf",
|
|
"*.docx",
|
|
"*.xlsx",
|
|
"*.pptx",
|
|
"*.iml",
|
|
".idea",
|
|
".vscode",
|
|
".DS_Store",
|
|
"Thumbs.db",
|
|
"venv",
|
|
"env",
|
|
]
|
|
|
|
[tool.mypy]
|
|
show_error_code_links = true
|
|
pretty = true
|
|
show_absolute_path = true
|
|
show_error_context = true
|
|
show_error_end = true
|
|
strict = true
|
|
warn_unreachable = true
|
|
install_types = true
|
|
plugins = ["pydantic.mypy"]
|
|
|
|
[[tool.mypy.overrides]]
|
|
module = ["jedi", "jedi.api.classes", "inquirer", "inquirer.themes", "numba"]
|
|
ignore_missing_imports = true
|
|
|
|
[tool.pydantic-mypy]
|
|
init_forbid_extra = true
|
|
init_typed = true
|
|
warn_required_dynamic_aliases = true
|
|
|
|
[tool.ruff]
|
|
target-version = "py39"
|
|
line-length = 120
|
|
fix = true
|
|
show-fixes = true
|
|
extend-exclude = ["code_to_optimize/", "pie_test_set/", "tests/", "experiments/"]
|
|
|
|
[tool.ruff.lint]
|
|
select = ["ALL"]
|
|
ignore = [
|
|
"N802",
|
|
"C901",
|
|
"D100",
|
|
"D101",
|
|
"D102",
|
|
"D103",
|
|
"D105",
|
|
"D107",
|
|
"D203", # incorrect-blank-line-before-class (incompatible with D211)
|
|
"D213", # multi-line-summary-second-line (incompatible with D212)
|
|
"S101",
|
|
"S603",
|
|
"S607",
|
|
"COM812",
|
|
"FIX002",
|
|
"PLR0912",
|
|
"PLR0913",
|
|
"PLR0915",
|
|
"TD002",
|
|
"TD003",
|
|
"TD004",
|
|
"PLR2004",
|
|
"UP007", # remove once we drop 3.9 support.
|
|
"E501",
|
|
"BLE001",
|
|
"ERA001",
|
|
"TRY003",
|
|
"EM101",
|
|
"T201",
|
|
"PGH004",
|
|
"S301",
|
|
"D104",
|
|
"PERF203",
|
|
"LOG015",
|
|
"PLC0415",
|
|
"UP045",
|
|
"TD007",
|
|
"D417",
|
|
"D401",
|
|
"S110", # try-except-pass - we do this a lot
|
|
"ARG002", # Unused method argument
|
|
# Added for multi-language branch
|
|
"FBT001", # Boolean positional argument
|
|
"FBT002", # Boolean default positional argument
|
|
"ANN401", # typing.Any disallowed
|
|
"ARG001", # Unused function argument (common in abstract/interface methods)
|
|
"TRY300", # Consider moving to else block
|
|
"TRY401", # Redundant exception in logging.exception
|
|
"PLR0911", # Too many return statements
|
|
"PLW0603", # Global statement
|
|
"PLW2901", # Loop variable overwritten
|
|
"SIM102", # Nested if statements
|
|
"SIM103", # Return negated condition
|
|
"ANN001", # Missing type annotation
|
|
"PLC0206", # Dictionary items
|
|
"S314", # XML parsing (acceptable for dev tool)
|
|
"S608", # SQL injection (internal use only)
|
|
"S112", # try-except-continue
|
|
"PERF401", # List comprehension suggestion
|
|
"SIM108", # Ternary operator suggestion
|
|
"F841", # Unused variable (often intentional)
|
|
"ANN202", # Missing return type for private functions
|
|
]
|
|
|
|
[tool.ruff.lint.flake8-type-checking]
|
|
strict = true
|
|
runtime-evaluated-base-classes = ["pydantic.BaseModel"]
|
|
runtime-evaluated-decorators = ["pydantic.validate_call", "pydantic.dataclasses.dataclass"]
|
|
|
|
[tool.ruff.lint.pep8-naming]
|
|
classmethod-decorators = [
|
|
# Allow Pydantic's `@validator` decorator to trigger class method treatment.
|
|
"pydantic.validator",
|
|
]
|
|
|
|
[tool.ruff.lint.isort]
|
|
split-on-trailing-comma = false
|
|
|
|
[tool.ruff.format]
|
|
docstring-code-format = true
|
|
skip-magic-trailing-comma = true
|
|
|
|
[tool.hatch.version]
|
|
source = "uv-dynamic-versioning"
|
|
|
|
[tool.uv]
|
|
workspace = { members = ["codeflash-benchmark"] }
|
|
|
|
[tool.uv.sources]
|
|
codeflash-benchmark = { workspace = true }
|
|
|
|
[tool.uv-dynamic-versioning]
|
|
enable = true
|
|
style = "pep440"
|
|
vcs = "git"
|
|
|
|
[tool.hatch.build.hooks.version]
|
|
path = "codeflash/version.py"
|
|
template = """# These version placeholders will be replaced by uv-dynamic-versioning during build.
|
|
__version__ = "{version}"
|
|
"""
|
|
|
|
|
|
#[tool.hatch.build.hooks.custom]
|
|
#path = "codeflash/update_license_version.py"
|
|
|
|
|
|
[tool.codeflash]
|
|
# All paths are relative to this pyproject.toml's directory.
|
|
module-root = "codeflash"
|
|
tests-root = "codeflash"
|
|
benchmarks-root = "tests/benchmarks"
|
|
ignore-paths = []
|
|
formatter-cmds = ["disabled"]
|
|
|
|
[tool.pytest.ini_options]
|
|
filterwarnings = [
|
|
"ignore::pytest.PytestCollectionWarning",
|
|
]
|
|
markers = [
|
|
"ci_skip: mark test to skip in CI environment",
|
|
]
|
|
|
|
|
|
[build-system]
|
|
requires = ["hatchling", "uv-dynamic-versioning"]
|
|
build-backend = "hatchling.build"
|
|
|