pyproject.toml

[build-system]
requires = ["setuptools", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "minolmo"
version = "0.1.0"
description = "A Simplified Version of OLMo Model"
authors = [
    { name = "Allen Institute for Artificial Intelligence", email = "olmo@allenai.org" },
    { name = "Research and Engineering at Kempner Institute", email = "kempner-research-engineering@g.harvard.edu" },
]
readme = "README.md"
requires-python = ">=3.10"
license = { file = "LICENSE" }
dependencies = [
    "numpy",
    "torch",
    "omegaconf",
    "rich",
    "requests",
]

[tool.setuptools.packages.find]
where = ["src"]

[project.optional-dependencies]
dev = [
    "ruff",
    "mypy>=1.0,<1.4",
    "black>=23.1,<24.0",
    "isort>=5.12,<5.13",
    "pytest",
    "pytest-sphinx",
    "twine>=1.11.0",
    "setuptools",
    "wheel",
    "build",
]
train = [
    "wandb",
    "beaker-gantry",
    "click",
    "torchmetrics",
    "smashed[remote]>=0.21.1",
    "safetensors",
    "datasets",
    "scikit-learn",
    "msgspec>=0.14.0",
]
all = [
    "minolmo[dev,train]",
]

docs = [
    "sphinx",
    "sphinx-rtd-theme",
    "sphinxcontrib-napoleon",
    "myst-parser",
    "nbsphinx",
    "recommonmark",
    "sphinx-copybutton",
    "sphinxcontrib-bibtex",

]

[project.urls]
Homepage = "https://github.com/KempnerInstitute/min-olmo"
Repository = "https://github.com/KempnerInstitute/min-olmo"

[tool.setuptools.dynamic]
version = { attr = "minolmo.version.VERSION" }

[tool.pytest.ini_options]
log_format = "%(asctime)s - %(levelname)s - %(name)s - %(message)s"
log_level = "DEBUG"