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

Cast to int

Browse source
This commit is contained in:
Oleksandr Shchur 2026-02-18 14:56:11 +00:00
parent 1949435850
commit b2020fa545
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/chronos/chronos2/dataset.py
View file

@ -536,9 +536,9 @@ class Chronos2Dataset(IterableDataset):
task = self.tasks[task_idx]
task_past_tensor = task["context"].clone()
task_future_tensor = task["future_covariates"].clone()
task_n_targets = task["n_targets"]
task_n_covariates = task["n_covariates"]
task_n_future_covariates = task["n_future_covariates"]
task_n_targets = int(task["n_targets"])
task_n_covariates = int(task["n_covariates"])
task_n_future_covariates = int(task["n_future_covariates"])
task_n_past_only_covariates = task_n_covariates - task_n_future_covariates
full_length = task_past_tensor.shape[-1]