timesfm/tests/__init__.py at 6ed1d8a7a28d07e5175e44f98fefb075300a4b62 - timesfm - Gitea: Git with a cup of tea

freedak/timesfm

Files

T

darkpowerxo c10494a4c5 test: add unit tests for configs, torch layers, utils, and base utils

Apply changes from PR #394 by @cj-wong:
- tests/__init__.py: package marker
- tests/test_base_utils.py: strip_leading_nans + linear_interpolation tests
- tests/test_configs.py: frozen dataclass, defaults, replace, equality tests
- tests/test_torch_layers.py: ResidualBlock, RMSNorm, RandomFourierFeatures
- tests/test_torch_utils.py: update_running_stats, revin, DecodeCache tests

2026-04-08 14:13:37 -04:00

2 lines

21 B

Python

Raw Blame History

# Tests for TimesFM.