summaryrefslogtreecommitdiff
path: root/pyproject.toml
blob: d9511ec6d3f6dfcfbe7809e077cf54c389dd6064 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
[project]
name = "dagformer"
version = "0.1.0"
description = "Context-conditioned DAG topology predictor for OLMo2-1B"
requires-python = ">=3.9"
dependencies = [
    "torch>=2.2",
    "transformers>=4.40",
    "datasets",
    "wandb",
    "pyyaml",
    "einops",
]

[build-system]
requires = ["setuptools>=64"]
build-backend = "setuptools.backends._legacy:_Backend"

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