Elgato_dark/chronos-forecasting
1
0
Fork 0
mirror of https://github.com/amazon-science/chronos-forecasting synced 2026-05-24 01:58:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Bump version from 2.2.0rc4 to 2.2.0 (#426)

Browse source 
*Issue #, if available:*

*Description of changes:*


By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice.
This commit is contained in:
Abdul Fatir 2025-12-08 13:49:34 +01:00 committed by GitHub
parent 5f5da1ac89
commit fdda16fc02
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
scripts/evaluation/evaluate.py
View file

@ -295,7 +295,7 @@ def chronos_2(
device: str = "cuda",
torch_dtype: str = "float32",
batch_size: int = 32,
predict_batches_jointly: bool = False,
cross_learning: bool = False,
):
"""Evaluate Chronos-2 models.
@ -316,7 +316,7 @@ def chronos_2(
batch_size : int, optional, default = 32
Batch size for inference. For Chronos-Bolt models, significantly larger
batch sizes can be used
predict_batches_jointly: bool, optional, default = False
cross_learning: bool, optional, default = False
If True, cross-learning is enables and model makes joint predictions for all
items in the batch
"""
@ -335,7 +335,7 @@ def chronos_2(
metrics_path=metrics_path,
model_id=model_id,
batch_size=batch_size,
predict_batches_jointly=predict_batches_jointly,
cross_learning=cross_learning,
)

2
src/chronos/__about__.py
View file

@ -1 +1 @@
__version__ = "2.2.0rc4"
__version__ = "2.2.0"