chronos-forecasting/test
Abdul Fatir efb86e02c2
Chronos-2: Add after_batch callback (#436)
*Issue #, if available:*

*Description of changes:* Adds support for custom callbacks after each
batch is processed during prediction. This allows for keeping track of
the time limit in AutoGluon.


By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice.
2025-12-17 11:45:16 +01:00
..
dummy-chronos-bolt-model Add support for Chronos-Bolt models (#204) 2024-11-26 17:47:14 +01:00
dummy-chronos-model Upload code 2024-03-13 09:58:39 +01:00
dummy-chronos2-lora Chronos-2: Add LoRA fine-tuning support (#393) 2025-11-25 16:06:58 +01:00
dummy-chronos2-model Add Chronos-2 (#319) 2025-10-20 10:34:20 +02:00
__init__.py Fix padding for int contexts (#227) 2024-12-04 16:46:17 +01:00
test_chronos.py Add predict_df support for Chronos and Chronos-Bolt (#371) 2025-11-11 18:37:19 +01:00
test_chronos2.py Chronos-2: Add after_batch callback (#436) 2025-12-17 11:45:16 +01:00
test_chronos_bolt.py Add predict_df support for Chronos and Chronos-Bolt (#371) 2025-11-11 18:37:19 +01:00
test_df_utils.py Add unittests for df_utils (#414) 2025-12-08 15:02:44 +00:00
test_utils.py Add Chronos-2 (#319) 2025-10-20 10:34:20 +02:00
util.py Chronos-2: Add after_batch callback (#436) 2025-12-17 11:45:16 +01:00