[pull] master from polakowo:master

.gitattributes

@@ -1 +1,2 @@
-*.ipynb linguist-language=Python
+*.ipynb linguist-language=Python
+*.rs linguist-language=Rust

.github/workflows/docker.yml

@@ -1,9 +1,26 @@
 name: Docker
 
 on:
-  release:
-    types: [published]
+  workflow_call:
+    inputs:
+      ref:
+        required: false
+        type: string
+      prerelease:
+        required: false
+        type: boolean
+        default: false
   workflow_dispatch:
+    inputs:
+      ref:
+        description: "Git ref to build. Defaults to the workflow ref."
+        required: false
+        type: string
+      prerelease:
+        description: "Skip latest tags."
+        required: false
+        type: boolean
+        default: false
 
 permissions:
   contents: read
@@ -23,16 +40,26 @@ jobs:
             tag_suffix: ""
             latest_tag: latest
 
+          - variant: rust
+            vbt_extras: "rust"
+            tag_suffix: "-rust"
+            latest_tag: latest-rust
+
           - variant: full
             vbt_extras: "full"
             tag_suffix: "-full"
             latest_tag: latest-full
 
+          - variant: full-rust
+            vbt_extras: "full,rust"
+            tag_suffix: "-full-rust"
+            latest_tag: latest-full-rust
+
     steps:
       - name: Checkout release tag
         uses: actions/checkout@v4
         with:
-          ref: ${{ github.event.release.tag_name || github.ref }}
+          ref: ${{ inputs.ref || github.ref }}
           fetch-depth: 1
           fetch-tags: true
           filter: blob:none
@@ -58,7 +85,7 @@ jobs:
           tags: |
             type=ref,event=tag,suffix=${{ matrix.tag_suffix }}
             type=semver,pattern={{version}}${{ matrix.tag_suffix }}
-            type=raw,value=${{ matrix.latest_tag }},enable=${{ !github.event.release.prerelease }}
+            type=raw,value=${{ matrix.latest_tag }},enable=${{ !inputs.prerelease }}
 
       - name: Build and push (multi-arch) - ${{ matrix.variant }}
         uses: docker/build-push-action@v6
@@ -72,4 +99,4 @@ jobs:
           build-args: |
             VBT_EXTRAS=${{ matrix.vbt_extras }}
           cache-from: type=gha
-          cache-to: type=gha,mode=max
+          cache-to: type=gha,mode=max

.github/workflows/pypi.yml

@@ -4,12 +4,18 @@ on:
   release:
     types: [published]
   workflow_dispatch:
+    inputs:
+      dry_run:
+        description: "Build and upload artifacts, but skip PyPI and Docker publishing."
+        required: false
+        type: boolean
+        default: true
 
 permissions:
   contents: read
 
 jobs:
-  release-build:
+  vectorbt-build:
     runs-on: ubuntu-latest
 
     steps:
@@ -28,34 +34,193 @@ jobs:
 
       - name: Build release distributions
         run: |
-          # NOTE: put your own distribution build steps here.
           python -m pip install build
           python -m build
 
-      - name: Upload distributions
+      - name: Upload vectorbt distributions
         uses: actions/upload-artifact@v4
         with:
-          name: release-dists
+          name: vectorbt-dists
           path: dist/
 
-  pypi-publish:
+  rust-sdist:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout release tag
+        uses: actions/checkout@v4
+        with:
+          ref: ${{ github.event.release.tag_name || github.ref }}
+          fetch-depth: 1
+          fetch-tags: true
+          filter: blob:none
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+
+      - name: Build vectorbt-rust source distribution
+        run: |
+          python -m pip install maturin
+          python -m maturin sdist --manifest-path rust/Cargo.toml --out dist
+
+      - name: Upload vectorbt-rust source distribution
+        uses: actions/upload-artifact@v4
+        with:
+          name: vectorbt-rust-sdist
+          path: dist/
+
+  rust-wheels:
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        include:
+          - os: ubuntu-latest
+            target: x86_64-unknown-linux-gnu
+            python-version: "3.10"
+          - os: ubuntu-latest
+            target: x86_64-unknown-linux-gnu
+            python-version: "3.11"
+          - os: ubuntu-latest
+            target: x86_64-unknown-linux-gnu
+            python-version: "3.12"
+          - os: ubuntu-latest
+            target: x86_64-unknown-linux-gnu
+            python-version: "3.13"
+          - os: ubuntu-latest
+            target: aarch64-unknown-linux-gnu
+            python-version: "3.10"
+          - os: ubuntu-latest
+            target: aarch64-unknown-linux-gnu
+            python-version: "3.11"
+          - os: ubuntu-latest
+            target: aarch64-unknown-linux-gnu
+            python-version: "3.12"
+          - os: ubuntu-latest
+            target: aarch64-unknown-linux-gnu
+            python-version: "3.13"
+          - os: macos-latest
+            target: x86_64-apple-darwin
+            python-version: "3.10"
+          - os: macos-latest
+            target: x86_64-apple-darwin
+            python-version: "3.11"
+          - os: macos-latest
+            target: x86_64-apple-darwin
+            python-version: "3.12"
+          - os: macos-latest
+            target: x86_64-apple-darwin
+            python-version: "3.13"
+          - os: macos-latest
+            target: aarch64-apple-darwin
+            python-version: "3.10"
+          - os: macos-latest
+            target: aarch64-apple-darwin
+            python-version: "3.11"
+          - os: macos-latest
+            target: aarch64-apple-darwin
+            python-version: "3.12"
+          - os: macos-latest
+            target: aarch64-apple-darwin
+            python-version: "3.13"
+          - os: windows-latest
+            target: x86_64-pc-windows-msvc
+            python-version: "3.10"
+          - os: windows-latest
+            target: x86_64-pc-windows-msvc
+            python-version: "3.11"
+          - os: windows-latest
+            target: x86_64-pc-windows-msvc
+            python-version: "3.12"
+          - os: windows-latest
+            target: x86_64-pc-windows-msvc
+            python-version: "3.13"
+
+    steps:
+      - name: Checkout release tag
+        uses: actions/checkout@v4
+        with:
+          ref: ${{ github.event.release.tag_name || github.ref }}
+          fetch-depth: 1
+          fetch-tags: true
+          filter: blob:none
+
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Build vectorbt-rust wheel
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.target }}
+          args: --release --locked --manifest-path rust/Cargo.toml --out dist --compatibility pypi -i python${{ matrix.python-version }}
+          sccache: "true"
+          manylinux: auto
+
+      - name: Upload vectorbt-rust wheel
+        uses: actions/upload-artifact@v4
+        with:
+          name: vectorbt-rust-wheel-${{ matrix.target }}-py${{ matrix.python-version }}
+          path: dist/
+
+  vectorbt-rust-publish:
+    if: ${{ !inputs.dry_run }}
+    runs-on: ubuntu-latest
+    needs:
+      - rust-sdist
+      - rust-wheels
+    permissions:
+      id-token: write
+
+    environment:
+      name: pypi
+
+    steps:
+      - name: Retrieve vectorbt-rust distributions
+        uses: actions/download-artifact@v4
+        with:
+          pattern: vectorbt-rust-*
+          path: dist/
+          merge-multiple: true
+
+      - name: Publish vectorbt-rust distributions to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          packages-dir: dist/
+
+  vectorbt-publish:
+    if: ${{ !inputs.dry_run }}
     runs-on: ubuntu-latest
     needs:
-      - release-build
+      - vectorbt-build
+      - vectorbt-rust-publish
     permissions:
       id-token: write
 
     environment:
       name: pypi
 
     steps:
-      - name: Retrieve release distributions
+      - name: Retrieve vectorbt distributions
         uses: actions/download-artifact@v4
         with:
-          name: release-dists
+          name: vectorbt-dists
           path: dist/
 
-      - name: Publish release distributions to PyPI
+      - name: Publish vectorbt distributions to PyPI
         uses: pypa/gh-action-pypi-publish@release/v1
         with:
           packages-dir: dist/
+
+  docker-publish:
+    if: ${{ !inputs.dry_run }}
+    needs:
+      - vectorbt-publish
+    uses: ./.github/workflows/docker.yml
+    with:
+      ref: ${{ github.event.release.tag_name || github.ref }}
+      prerelease: ${{ github.event.release.prerelease || false }}
+    secrets: inherit

.github/workflows/tests.yml

@@ -29,8 +29,43 @@ jobs:
       - name: Set up Python ${{ matrix.python-version }}
         run: uv python install ${{ matrix.python-version }}
 
+      - name: Create virtual environment
+        run: uv venv --python ${{ matrix.python-version }}
+
       - name: Install dependencies
-        run: uv sync --extra cov
+        run: uv pip install --python .venv -e ".[test]"
 
       - name: Run tests
-        run: uv run pytest tests/
+        run: uv run --python .venv --no-sync pytest tests/
+
+  rust-engine:
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest, windows-latest, macos-latest]
+        python-version: ["3.10", "3.11", "3.12", "3.13"]
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Install Rust
+        uses: dtolnay/rust-toolchain@stable
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v5
+
+      - name: Set up Python ${{ matrix.python-version }}
+        run: uv python install ${{ matrix.python-version }}
+
+      - name: Create virtual environment
+        run: uv venv --python ${{ matrix.python-version }}
+
+      - name: Install dependencies
+        run: uv pip install --python .venv -e ".[test-rust]"
+
+      - name: Build local Rust engine
+        run: uv run --python .venv --no-sync python -m maturin develop --manifest-path rust/Cargo.toml --release
+
+      - name: Run Rust engine tests
+        run: uv run --python .venv --no-sync pytest tests/