pyproject.toml

[project]
name = "autora-doc"
license = { file = "LICENSE" }
readme = "README.md"
authors = [{ name = "Carlos Garcia Jurado Suarez", email = "carlosg@uw.edu" }]
requires-python = ">=3.8"
classifiers = [
    "Development Status :: 4 - Beta",
    "License :: OSI Approved :: MIT License",
    "Intended Audience :: Developers",
    "Intended Audience :: Science/Research",
    "Operating System :: OS Independent",
    "Programming Language :: Python",
]
dynamic = ["version"]
dependencies = [
    "typer",
    "scipy",
    # This works, while installing from pytorch and cuda from conda does not",
    "torch==2.0.1",
    "transformers>=4.37.2",
]

# On a mac, install optional dependencies with `pip install '.[dev]'` (include the single quotes)
description = "Automatic documentation generator from AutoRA code"
[project.optional-dependencies]
dev = [
    "pytest",
    "pytest-cov",       # Used to report total code coverage
    "pre-commit",       # Used to run checks before finalizing a git commit
    "sphinx",           # Used to automatically generate documentation
    "sphinx-rtd-theme", # Used to render documentation
    "sphinx-autoapi",   # Used to automatically generate api documentation
    "black",            # Used for static linting of files
    "mypy",             # Used for static type checking of files
    # if you add dependencies here while experimenting in a notebook and you
    # want that notebook to render in your documentation, please add the
    # dependencies to ./docs/requirements.txt as well.
    "nbconvert",   # Needed for pre-commit check to clear output from Python notebooks
    "nbsphinx",    # Used to integrate Python notebooks into Sphinx documentation
    "ipython",     # Also used in building notebooks into Sphinx
    "matplotlib",  # Used in sample notebook intro_notebook.ipynb
    "ipykernel",
    "hf_transfer",
]
pipelines = [
    "jsonlines",
    "mlflow",
    "nltk",
    "sentence-transformers>=2.3.1",
    "peft>=0.8.2",
    "trl>=0.7.10",
    "datasets",
    "tensorboardX",
]
# NOTE: When updating dependencies, in particular cuda/azure ml, make sure to update the azureml/conda.yaml too
azure = ["azureml-core", "azureml-mlflow"]
cuda = ["bitsandbytes>=0.42.0", "accelerate>=0.24.1", "xformers"]

[project.urls]
Homepage = "https://github.com/AutoResearch/autodoc"

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

[tool.pytest.ini_options]
testpaths = ["tests"]

[tool.black]
line-length = 110
target-version = ["py38"]

[tool.isort]
profile = "black"
line_length = 110

[tool.coverage.run]
omit = ["src/autora/doc/_version.py"]

[tool.hatch]


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

[tool.hatch.build.hooks.vcs]
version-file = "src/autora/doc/_version.py"

[tool.hatch.version.raw-options]
local_scheme = "no-local-version"

[tool.hatch.build.targets.sdist]
include = ["src/autora"]

[tool.hatch.build.targets.wheel]
packages = ["src/autora"]

[project.scripts]
autodoc = "autora.doc.pipelines.main:app"