Elgato_dark/chronos-forecasting
1
0
Fork 0
mirror of https://github.com/amazon-science/chronos-forecasting synced 2026-05-23 17:48:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Simplify test

Browse source
This commit is contained in:
Oleksandr Shchur 2026-02-18 15:55:12 +00:00
parent c93fadc57c
commit 588f19f1f8
1 changed files with 1 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
test/test_chronos2.py
View file

@ -1154,13 +1154,7 @@ def test_pipeline_can_be_finetuned_with_preprocessed_hf_dataset(pipeline):
# Preprocess and convert to HF Dataset (simulating Arrow-based lazy loading)
prepared_tasks = prepare_tasks(raw_inputs, prediction_length=prediction_length, min_past=1, mode="train")
hf_dataset = datasets.Dataset.from_dict({
"context": [task["context"].numpy() for task in prepared_tasks],
"future_covariates": [task["future_covariates"].numpy() for task in prepared_tasks],
"n_targets": [task["n_targets"] for task in prepared_tasks],
"n_covariates": [task["n_covariates"] for task in prepared_tasks],
"n_future_covariates": [task["n_future_covariates"] for task in prepared_tasks],
}).with_format("torch")
hf_dataset = datasets.Dataset.from_list(prepared_tasks).with_format("torch")
# Fine-tune with preprocessed inputs
ft_pipeline = pipeline.fit(