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. """Options for forecasting.
Attributes: 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 function at inference time during batched inference. Any input time series
with length less than max_context will be padded with zeros, and with with length less than max_context will be padded with zeros, and with
length greater than max_context will be truncated. 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 function at inference time during batched inference. The compiled cached
decoding function will by default forecast till max_horizon. decoding function will by default forecast till max_horizon.
normalize_inputs: Whether to normalize the inputs. This is useful when the 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: if "torch_compile" in kwargs:
torch_compile = kwargs["torch_compile"] torch_compile = kwargs["torch_compile"]
if torch_compile: if torch_compile:
print("Compiling model...") logging.info("Compiling model...")
self = torch.compile(self) self = torch.compile(self)
self.eval() self.eval()