Skip to content

Conversation

@abdulfatir
Copy link
Contributor

Issue #, if available:

Description of changes: This PR adds predict_df to the base pipeline which enables pandas support for the univariate Chronos and Chronos-Bolt models.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@abdulfatir abdulfatir requested a review from lostella November 7, 2025 00:23
@abdulfatir abdulfatir changed the title Add predict_df support for univariate models (Chronos and Chronos-Bolt) Add predict_df support for Chronos and Chronos-Bolt Nov 7, 2025
Copy link
Contributor

@canerturkmen canerturkmen left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! Dropped some comments.

f"Expected `target` to be str, but found {type(target)}. {self.__class__.__name__} only supports univariate forecasting."
)

if prediction_length is None:
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

again, if there is any way at all of reducing the repeat in Chronos2 I think we should consider it.

@abdulfatir abdulfatir added the run-eval Run evaluation CI workflow label Nov 7, 2025
Copy link
Contributor

@canerturkmen canerturkmen left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@abdulfatir abdulfatir merged commit e48f480 into amazon-science:main Nov 11, 2025
6 checks passed
@abdulfatir abdulfatir deleted the univar-predict-df branch November 11, 2025 17:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

run-eval Run evaluation CI workflow

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants