fix: correct 'complied' typo and replace print with logging

Apply changes from PR #396 by @shahrukhx01:
- Fix typo 'complied' -> 'compiled' in ForecastConfig docstrings
- Replace bare print() with logging.info() in load_checkpoint()
This commit is contained in:
darkpowerxo
2026-04-08 14:09:14 -04:00
parent b6ac2b3559
commit bc03b77e9e
2 changed files with 3 additions and 3 deletions
+2 -2
View File
@@ -23,11 +23,11 @@ class ForecastConfig:
"""Options for forecasting.
Attributes:
max_context: The maximum context length. This is used by the complied decode
max_context: The maximum context length. This is used by the compiled decode
function at inference time during batched inference. Any input time series
with length less than max_context will be padded with zeros, and with
length greater than max_context will be truncated.
max_horizon: The maximum horizon length. This is used by the complied decode
max_horizon: The maximum horizon length. This is used by the compiled decode
function at inference time during batched inference. The compiled cached
decoding function will by default forecast till max_horizon.
normalize_inputs: Whether to normalize the inputs. This is useful when the
+1 -1
View File
@@ -85,7 +85,7 @@ class TimesFM_2p5_200M_torch_module(nn.Module):
if "torch_compile" in kwargs:
torch_compile = kwargs["torch_compile"]
if torch_compile:
print("Compiling model...")
logging.info("Compiling model...")
self = torch.compile(self)
self.eval()