mirror of
https://github.com/amazon-science/chronos-forecasting
synced 2026-05-23 09:39:35 +00:00
84 lines
2.1 KiB
TOML
84 lines
2.1 KiB
TOML
[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",
|
|
"typing_extensions~=4.0",
|
|
]
|
|
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`
|
|
]
|