Skip to content

feat: add tirex-2 model - #366

Merged
AzulGarza merged 11 commits into
mainfrom
feat/tirex-2
Aug 10, 2026
Merged

feat: add tirex-2 model#366
AzulGarza merged 11 commits into
mainfrom
feat/tirex-2

Conversation

@AzulGarza

Copy link
Copy Markdown
Member

this pr adds tirex-2, including its support trough the Tirex class, tests, and an example. in prepartion for a new release, it also adds a changelog and a new structure for the examples section.

@cursor

cursor Bot commented Aug 10, 2026

Copy link
Copy Markdown

Bugbot is not enabled for your account, so this pull request was not reviewed.

Enable Bugbot in the Cursor dashboard to get automatic reviews on future PRs.

Copilot AI left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

Adds TiRex-2 support to the existing TiRex forecaster, updates project dependencies/tests accordingly, and restructures the documentation Examples section while introducing a new v0.0.30 changelog entry.

Changes:

  • Add TiRex-2 backend dispatching in TiRex (model loading + forecasting path).
  • Add TiRex-2 tests and include TiRex-2 in the shared model test matrix.
  • Restructure docs navigation/examples and add a new changelog page for v0.0.30.

Reviewed changes

Copilot reviewed 12 out of 14 changed files in this pull request and generated 1 comment.

Show a summary per file
File Description
timecopilot/models/foundation/tirex.py Adds TiRex-2 detection + v1/v2 model loading and forecasting dispatch.
tests/models/foundation/test_tirex.py Adds basic dispatch and TiRex-2 forecasting tests.
tests/models/conftest.py Adds TiRex-2 to the shared model list for model-level tests.
pyproject.toml Adds TiRex-2-related dependencies and bumps timecopilot-uni2ts.
mkdocs.yml Restructures Examples nav and adds v0.0.30 changelog entry.
docs/model-hub.md Updates model hub to reference family example notebooks and TiRex/TiRex-2 naming.
docs/getting-started/quickstart.md Updates Quickstart to point readers to the new Examples index.
docs/examples/index.md Adds a new Examples landing page organizing notebooks by category.
docs/contributing.md Updates contributor guidance to match the new Examples nav structure.
docs/changelogs/v0.0.30.md Adds v0.0.30 changelog entry describing TiRex-2 support and related changes.
docs/changelogs/index.md Adds v0.0.30 link to the changelog index.
.github/workflows/ci.yaml Updates lint step to run pre-commit via uv run.
Suppressed comments (1)

timecopilot/models/foundation/tirex.py:109

  • _get_model_v1 sets os.environ["TIREX_NO_CUDA"] = "1" when running on CPU, but never restores the previous value. Because this is a process-wide environment variable, it can leak into later model loads (including subsequent TiRex calls in the same process) and change behavior unexpectedly. Capture the previous value and restore it in the finally block.
        device = "cuda" if torch.cuda.is_available() else "cpu"
        if device == "cpu":
            # see https://github.com/NX-AI/tirex/tree/main?tab=readme-ov-file#cuda-kernels
            os.environ["TIREX_NO_CUDA"] = "1"
        model = load_model(self.repo_id, device=device)

💡 Add a code-review agent skill or configure MCP servers for context-aware, tailored reviews. Learn more in the docs.

Comment thread docs/changelogs/v0.0.30.md Outdated
@AzulGarza
AzulGarza merged commit c913413 into main Aug 10, 2026
@AzulGarza
AzulGarza deleted the feat/tirex-2 branch August 10, 2026 20:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants