feat: add toto-2

docs/changelogs/index.md

@@ -2,6 +2,7 @@
 
 Welcome to the TimeCopilot Changelog. Here, you will find a comprehensive list of all the changes, updates, and improvements made to the TimeCopilot project. This section is designed to keep you informed about the latest features, bug fixes, and enhancements as we continue to develop and refine the TimeCopilot experience. Stay tuned for regular updates and feel free to explore the details of each release below.
 
+- [v0.0.27](v0.0.27.md)
 - [v0.0.26](v0.0.26.md)
 - [v0.0.25](v0.0.25.md)
 - [v0.0.24](v0.0.24.md)

docs/changelogs/v0.0.27.md

@@ -0,0 +1,26 @@
+### Features
+
+* **Toto 2.0 support**: The `Toto` class now transparently supports both Toto 1.0 and Toto 2.0 foundation models. Pass a Toto 2.0 checkpoint (e.g. `Datadog/Toto-2.0-4m`) as `repo_id` and the model family is detected automatically from the checkpoint configuration. Toto 2.0 predicts a fixed set of quantile knots (0.1, ..., 0.9): the median is used as the point forecast and requested `quantiles`/`level` are obtained by linear interpolation across the knots. See [#343](https://github.com/TimeCopilot/timecopilot/pull/343).
+
+    ```python
+    import pandas as pd
+    from timecopilot.models.foundation.toto import Toto
+
+    df = pd.read_csv(
+        "https://timecopilot.s3.amazonaws.com/public/data/air_passengers.csv",
+        parse_dates=["ds"],
+    )
+
+    model = Toto(repo_id="Datadog/Toto-2.0-4m", alias="Toto-2")
+    fcst_df = model.forecast(df, h=12, quantiles=[0.1, 0.5, 0.9])
+    ```
+
+    Available Toto 2.0 checkpoints: `Datadog/Toto-2.0-4m`, `Datadog/Toto-2.0-22m`, `Datadog/Toto-2.0-313m`, `Datadog/Toto-2.0-1B`, and `Datadog/Toto-2.0-2.5B`.
+
+### Documentation
+
+* **Toto family example**: Added the [Toto Family](https://timecopilot.dev/examples/toto-family/) notebook, comparing Toto 1.0, Toto 2.0, `Prophet`, `AutoARIMA`, and `SeasonalNaive`. See [#343](https://github.com/TimeCopilot/timecopilot/pull/343).
+
+---
+
+**Full Changelog**: https://github.com/TimeCopilot/timecopilot/compare/v0.0.26...v0.0.27

mkdocs.yml

@@ -28,6 +28,7 @@ nav:
           - examples/ts-foundation-models-comparison-quickstart.ipynb
           - examples/gift-eval.ipynb
           - examples/chronos-family.ipynb
+          - examples/toto-family.ipynb
           - examples/finetuning.ipynb
           - examples/cryptocurrency-quickstart.ipynb
           - examples/sktime.ipynb
@@ -54,6 +55,9 @@ nav:
       - api/gift-eval/gift-eval.md
   - Changelogs:
       - changelogs/index.md
+      - changelogs/v0.0.27.md
+      - changelogs/v0.0.26.md
+      - changelogs/v0.0.25.md
       - changelogs/v0.0.24.md
       - changelogs/v0.0.23.md
       - changelogs/v0.0.22.md

pyproject.toml

@@ -96,7 +96,8 @@ dependencies = [
   "timecopilot-granite-tsfm>=0.1.2",
   "timecopilot-timesfm>=0.2.1",
   "timecopilot-tirex>=0.1.0 ; python_full_version >= '3.11'",
-  "timecopilot-toto>=0.1.5",
+  "timecopilot-toto-2>=0.1.1",
+  "timecopilot-toto>=0.1.6",
   "timecopilot-uni2ts>=0.1.2 ; python_full_version < '3.14'",
   "torchmetrics>=1.8.2",
   "transformers>=4.41,<5 ; python_full_version < '3.13'",

tests/models/conftest.py

@@ -108,6 +108,12 @@ def disable_mps_session(monkeypatch):
         context_length=2_048,
     ),
     Toto(context_length=256, batch_size=2),
+    Toto(
+        repo_id="Datadog/Toto-2.0-4m",
+        alias="Toto-2",
+        context_length=256,
+        batch_size=2,
+    ),
     Moirai(
         context_length=256,
         batch_size=2,