From c04fd737261bd04e4f86c819a87cd81638b9a8ab Mon Sep 17 00:00:00 2001 From: Abdul Fatir Ansari Date: Fri, 31 Oct 2025 00:00:54 +0100 Subject: [PATCH 1/4] Add Chronos as a library --- .../tasks/src/model-libraries-snippets.ts | 24 +++++++++++++++++++ packages/tasks/src/model-libraries.ts | 7 ++++++ 2 files changed, 31 insertions(+) diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index 48ce6886ec..f6391cbe41 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -132,6 +132,30 @@ wav = model.generate(text, audio_prompt_path=AUDIO_PROMPT_PATH) ta.save("test-2.wav", wav, model.sr)`, ]; +export const chronos_forecasting = (model: ModelData): string[] => { + const installSnippet = `pip install chronos-forecasting`; + + const exampleSnippet = `import pandas as pd +from chronos import BaseChronosPipeline + +pipeline = BaseChronosPipeline.from_pretrained("${model.id}", device_map="cuda") + +# Load historical data +context_df = pd.read_csv("https://autogluon.s3.us-west-2.amazonaws.com/datasets/timeseries/misc/AirPassengers.csv") + +# Generate predictions +pred_df = pipeline.predict_df( + context_df, + prediction_length=36, # Number of steps to forecast + quantile_levels=[0.1, 0.5, 0.9], # Quantiles for probabilistic forecast + id_column="item_id", # Column identifying different time series + timestamp_column="Month", # Column with datetime information + target="#Passengers", # Column(s) with time series values to predict +)`; + + return [installSnippet, exampleSnippet]; +}; + export const contexttab = (): string[] => { const installSnippet = `pip install git+https://github.com/SAP-samples/contexttab`; diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 554b229a10..3b359b040e 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -229,6 +229,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { repoUrl: "https://github.com/Unbabel/COMET/", countDownloads: `path:"hparams.yaml"`, }, + chronos_forecasting: { + prettyLabel: "Chronos", + repoName: "chronos-forecasting", + repoUrl: "https://github.com/amazon-science/chronos-forecasting", + countDownloads: `path:"config.json"`, + snippets: snippets.chronos_forecasting, + }, contexttab: { prettyLabel: "ConTextTab", repoName: "ConTextTab", From 0db05c8376a94067cf3740b7462184a0f6c63951 Mon Sep 17 00:00:00 2001 From: Abdul Fatir Ansari Date: Fri, 31 Oct 2025 00:02:36 +0100 Subject: [PATCH 2/4] Fix --- packages/tasks/src/model-libraries.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 3b359b040e..b75ec2bf53 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -229,7 +229,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { repoUrl: "https://github.com/Unbabel/COMET/", countDownloads: `path:"hparams.yaml"`, }, - chronos_forecasting: { + "chronos-forecasting": { prettyLabel: "Chronos", repoName: "chronos-forecasting", repoUrl: "https://github.com/amazon-science/chronos-forecasting", From 16e2c989052ed8a5ee8b4b4fa023798e0bd9a5ce Mon Sep 17 00:00:00 2001 From: Abdul Fatir Date: Fri, 31 Oct 2025 10:37:17 +0100 Subject: [PATCH 3/4] Update packages/tasks/src/model-libraries.ts Co-authored-by: Lucain --- packages/tasks/src/model-libraries.ts | 1 - 1 file changed, 1 deletion(-) diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index b75ec2bf53..c8ca3782c7 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -233,7 +233,6 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { prettyLabel: "Chronos", repoName: "chronos-forecasting", repoUrl: "https://github.com/amazon-science/chronos-forecasting", - countDownloads: `path:"config.json"`, snippets: snippets.chronos_forecasting, }, contexttab: { From aad73e2b48e2c59a5eba103ed901530e5c7457c2 Mon Sep 17 00:00:00 2001 From: Abdul Fatir Ansari Date: Fri, 31 Oct 2025 10:42:04 +0100 Subject: [PATCH 4/4] Reorder --- packages/tasks/src/model-libraries.ts | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index c8ca3782c7..69b0db6223 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -209,6 +209,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { filter: false, countDownloads: `path:"asset/GPT.pt"`, }, + "chronos-forecasting": { + prettyLabel: "Chronos", + repoName: "Chronos", + repoUrl: "https://github.com/amazon-science/chronos-forecasting", + snippets: snippets.chronos_forecasting, + }, "cloud-agents": { prettyLabel: "Cloud Agents", repoName: "Cloud Agents", @@ -229,12 +235,6 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { repoUrl: "https://github.com/Unbabel/COMET/", countDownloads: `path:"hparams.yaml"`, }, - "chronos-forecasting": { - prettyLabel: "Chronos", - repoName: "chronos-forecasting", - repoUrl: "https://github.com/amazon-science/chronos-forecasting", - snippets: snippets.chronos_forecasting, - }, contexttab: { prettyLabel: "ConTextTab", repoName: "ConTextTab",