chronos-forecasting/pyproject.toml at main · amazon-science/chronos-forecasting · 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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
[project]
name = "chronos-forecasting"
dynamic = ["version"]
authors = [
  { name = "Abdul Fatir Ansari", email = "ansarnd@amazon.com" },
  { name = "Lorenzo Stella", email = "stellalo@amazon.com" },
  { name = "Caner Turkmen", email = "atturkm@amazon.com" },
  { name = "Oleksandr Shchur", email = "shchuro@amazon.com" },
  { name = "Jaris Küken", email = "kuekenj@amazon.com" },
  { name = "Andreas Auer", email = "auerand@amazon.com" },
]
description = "Chronos: Pretrained models for time series forecasting"
readme = "README.md"
license = { file = "LICENSE" }
requires-python = ">=3.10"
dependencies = [
  "torch>=2.2,<3",
  "transformers>=4.41,<5",
  "accelerate>=0.34,<2",
  "numpy>=1.21,<3",
  "einops>=0.7.0,<1",
  "scikit-learn>=1.6.0,<2",
]
classifiers = [
  "Programming Language :: Python :: 3",
  "License :: OSI Approved :: Apache Software License",
  "Operating System :: OS Independent",
  "Topic :: Scientific/Engineering :: Artificial Intelligence",
]

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

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

[tool.hatch.version]
path = "src/chronos/__about__.py"

[project.optional-dependencies]
extras = [
  "boto3>=1.10,<2",
  "peft>=0.13.0,<0.18",
  "fev>=0.6.1",
  "pandas[pyarrow]>=2.0,<2.4",
]
test = [
  "pytest~=8.0",
  "boto3>=1.10,<2",
  "peft>=0.13.0,<1",
  "fev>=0.6.1",
  "pandas[pyarrow]>=2.0,<2.4",
]
typecheck = ["mypy~=1.9"]
dev = [
  "gluonts[pro]~=0.16",
  "datasets>=3.0,<4",
  "typer",
  "tensorboard",
  "typer-config",
  "joblib",
  "fev>=0.6.1",
  "rich",
  "pandas>=2.0,<2.4",
]

[project.urls]
Homepage = "https://github.com/amazon-science/chronos-forecasting"
Issues = "https://github.com/amazon-science/chronos-forecasting/issues"
Chronos-Paper = "https://arxiv.org/abs/2403.07815"
Chronos-2-Paper = "https://arxiv.org/abs/2510.15821"

[tool.mypy]
ignore_missing_imports = true

[tool.ruff]
line-length = 119
lint.ignore = [
  "E501", # Line too long
  "E731", # Do not assign a `lambda` expression, use a `def`
  "E722", # Do not use bare `except`
]