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

fix version of datasets.load_dataset

Browse source
This commit is contained in:
pedroml 2025-09-05 12:51:21 +00:00
parent 563cb70f12
commit 2b52bfc500
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/chronosx/utils/hf_data_loader.py
View file

@ -76,7 +76,7 @@ def load_and_split_dataset(backtest_config: dict):
else:
ds = datasets.load_dataset(
hf_repo, dataset_name, split="train", trust_remote_code=True
hf_repo, dataset_name, split="train", trust_remote_code=True, revision="v0.1.0"
)
ds.set_format("numpy")