pyproject.toml

[project]
authors = [
  {name = "Yuxiang Wei", email = "yuxiang630.wei@gmail.com"},
]
dependencies = [
    "openai>=1.2.2",
    "transformers>=4.35.0",
    "torch>=2.1.0",
    "tiktoken>=0.5.1",
    "GitPython>=3.1.40",
    "datasets>=2.14.6",
    "numpy>=1.26.1",
    "sentence-transformers>=2.2.2",
]
description = "Self-instruct powered by seed code snippets from the wild"
license = {text = "MIT"}
name = "magicoder"
readme = "README.md"
requires-python = ">=3.10,<3.13"
version = "0.1.0"

[project.optional-dependencies]
flash-attn = [
    "packaging>=23.2",
    "flash-attn>=2.3.5",
    "wheel>=0.42.0",
]
[build-system]
build-backend = "pdm.backend"
requires = ["pdm-backend"]

[tool.pdm.dev-dependencies]
dev = [
    "ipython>=8.17.2",
    "pre-commit>=3.5.0",
    "accelerate>=0.24.1",
    "mypy>=1.7.0",
    "scikit-learn>=1.3.2",
    "matplotlib>=3.8.1",
    "InstructorEmbedding>=1.0.1",
    "joblib>=1.3.2",
    "-e file:///${PROJECT_ROOT}/../evalplus#egg=evalplus",
    "-e file:///${PROJECT_ROOT}/../bigcode-evaluation-harness#egg=bigcode-eval",
    "termcolor>=2.3.0",
    "treelib>=1.7.0",
    "tree-sitter>=0.20.4",
    "black>=23.11.0",
]
test = [
  "pytest>=7.4.3",
]

[tool.pytest.ini_options]
addopts = [
  "--import-mode=importlib",
]

[tool.black]
include = '\.pyi?$'
line-length = 88
target-version = ["py310"]

[tool.isort]
line_length = 88
profile = "black"
skip_gitignore = true

[tool.mypy]
check_untyped_defs = true
follow_imports = "silent"
ignore_missing_imports = true
# modules = ["main"]
mypy_path = "src"
packages = ["magicoder"]
python_version = "3.10"