chronos-forecasting/test
Lorenzo Stella 67f008432a
Fix padding for int contexts (#227)
*Issue #, if available:* On Linux, the final call to `.to` creates
trouble when input tensors are integer. For example:

```
>>> a = torch.tensor([1])
>>> b = torch.stack([torch.full((1,), torch.nan), a])
>>> b
tensor([[nan],
        [1.]])
>>> b.to(a)
tensor([[-9223372036854775808],
        [                   1]])
```


By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice.
2024-12-04 16:46:17 +01:00
..
dummy-chronos-bolt-model Add support for Chronos-Bolt models (#204) 2024-11-26 17:47:14 +01:00
dummy-chronos-model Upload code 2024-03-13 09:58:39 +01:00
__init__.py Fix padding for int contexts (#227) 2024-12-04 16:46:17 +01:00
test_chronos.py Return predictions in fp32 on CPU (#219) 2024-11-29 16:54:21 +01:00
test_chronos_bolt.py Return predictions in fp32 on CPU (#219) 2024-11-29 16:54:21 +01:00
test_utils.py Fix padding for int contexts (#227) 2024-12-04 16:46:17 +01:00
util.py Fix padding for int contexts (#227) 2024-12-04 16:46:17 +01:00