4 Commits

Author	SHA1	Message	Date
Rajat Sen	2f1625c208	Fix model loading issues and forecast_naive slicing bug in TimesFM 2.5 ... - Allow model wrapper constructors (__init__) to accept and ignore extra keyword arguments (e.g. proxies) passed by huggingface_hub during from_pretrained. - Implement load_checkpoint for TimesFM_2p5_200M_torch and TimesFM_2p5_200M_flax to restore weights from local paths. - Fix slicing bug in PyTorch's forecast_naive to correctly slice the time/horizon dimension ([:, :horizon, :]) instead of quantiles. - Add unit tests in tests/test_model_loading.py covering local checkpoint loading, hub compatibility, and prediction shape correctness.	2026-06-07 17:28:32 +00:00
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
siriuz42	7d8f3d971d	2.0.0 initial	2025-09-12 00:18:08 +00:00
tanmayshishodia	d4d4afd436	add pytest framework	2024-07-17 21:00:11 +00:00