Cents/pyproject.toml at main · DAI-Lab/Cents · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
[tool.poetry]
name = "cents-ml"
version = "1.0.1"
description = "A library for generating contextual timeseries data."
authors = ["Michael Fuest"]
license = "MIT License"
readme = "README.md"

packages = [
    { include = "cents" },
]

exclude = [
    "cents/outputs/**",
    "cents/checkpoints/**",
    "cents/data/**",
    "tests/**",
    "tutorials/**",
    ".github/**",
]

[tool.poetry.dependencies]

# Other dependencies
python = "^3.9"
torch = "2.6.0"
torchaudio = "2.6.0"
torchvision = "^0.21.0"
numpy = "^2.0.0"
pandas = "^2.2.3"
matplotlib = "^3.9.4"
scikit-learn = "^1.6.0"
tiktoken = "^0.8.0"
transformers = "^4.48.0"
accelerate = "^1.2.0"
pyyaml = "^6.0.2"
pre-commit = "^4.0.1"
mypy = "^1.13.0"
black = "^24.10.0"
isort = "^5.13.2"
autoflake = "^2.3.1"
dtaidistance = "^2.3.12"
seaborn = "^0.13.2"
einops = "^0.8.0"
sentencepiece = "^0.2.0"
omegaconf = "^2.3.0"
pytest = "^8.3.4"
hydra-core = "^1.3.2"
pytorch-lightning = "^2.4.0"
wandb = "^0.19.6"
pytest-cov = "^6.0.0"
botocore = "^1.36.24"
ipykernel = "^6.29.5"
jupyter = "^1.1.1"
huggingface-hub = "^0.32.0"


[tool.poetry.group.dev.dependencies]
pytest = "^8.3.4"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"