mirror of
https://github.com/amazon-science/chronos-forecasting
synced 2026-05-24 10:08:33 +00:00
1e102f6989
*Description of changes:* See title. By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice. Co-authored-by: Abdul Fatir Ansari <ansarnd@amazon.com>
18 lines
462 B
TOML
18 lines
462 B
TOML
[project]
|
|
name = "chronos"
|
|
version = "1.1.0"
|
|
requires-python = ">=3.8"
|
|
license = { file = "LICENSE" }
|
|
dependencies = [
|
|
"torch~=2.1", # package was tested on 2.2
|
|
"transformers~=4.31",
|
|
"accelerate",
|
|
]
|
|
|
|
[project.optional-dependencies]
|
|
test = ["pytest~=8.0", "numpy~=1.21"]
|
|
typecheck = ["mypy~=1.9"]
|
|
training = ["gluonts[pro]", "numpy", "tensorboard", "typer", "typer-config", "joblib", "scikit-learn"]
|
|
|
|
[tool.mypy]
|
|
ignore_missing_imports = true
|