From 7513dee92de065103a81efe2565d02aafa437f7e Mon Sep 17 00:00:00 2001 From: Abdul Fatir Date: Wed, 3 Dec 2025 15:33:08 +0100 Subject: [PATCH] Chronos-2: Do not catch exceptions in predict_fev when fine-tuning (#417) *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. --- src/chronos/chronos2/pipeline.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/src/chronos/chronos2/pipeline.py b/src/chronos/chronos2/pipeline.py index 9f8a0dc..37bb3c5 100644 --- a/src/chronos/chronos2/pipeline.py +++ b/src/chronos/chronos2/pipeline.py @@ -1030,11 +1030,7 @@ class Chronos2Pipeline(BaseChronosPipeline): finetune_kwargs["prediction_length"] = first_window.horizon finetune_kwargs["batch_size"] = finetune_kwargs.get("batch_size", batch_size) - try: - pipeline = self.fit(inputs=inputs, **finetune_kwargs) - except Exception as e: - msg = f"Finetuning failed with error: {e}. Continuing with the pretrained model." - warnings.warn(msg, category=UserWarning, stacklevel=2) + pipeline = self.fit(inputs=inputs, **finetune_kwargs) predictions_per_window = [] inference_time_s = 0.0