Skip to content

Use predict_df inside predict_fev#512

Open
shchur wants to merge 9 commits into
amazon-science:mainfrom
shchur:predict-fev-simplification
Open

Use predict_df inside predict_fev#512
shchur wants to merge 9 commits into
amazon-science:mainfrom
shchur:predict-fev-simplification

Conversation

@shchur

@shchur shchur commented Jun 17, 2026

Copy link
Copy Markdown
Contributor

Issue #, if available:

Description of changes:

  • Use predict_df + fev's built-in conversion utilities inside predict_fev for BaseChronosPipeline and Chronos2Pipeline.
  • Remove the bespoke fev-related logic in chronos.chronos2.dataset
  • Add batch_size support for BaseChronosPipeline.predict_df -> previously this method just ran inference on the entire DF in one go, potentially leading to OOM errors.

To do:

  • Run full benchmark for Chronos-2 and Chronos-Bolt to ensure that the results match v2.2.2

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

@shchur shchur force-pushed the predict-fev-simplification branch from 090cb40 to 85cd9eb Compare June 17, 2026 12:37
@shchur shchur requested a review from abdulfatir June 17, 2026 14:39
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant