From d7649b8457c02914e75198d32d7705ddc0c6dd20 Mon Sep 17 00:00:00 2001
From: Ivan Podkidyshev <ipodkidyshev@nvidia.com>
Date: Mon, 15 Jun 2026 16:02:59 +0200
Subject: [PATCH 1/2] use uv in ci

---
 .github/workflows/ci.yml | 21 +++++++++++++--------
 1 file changed, 13 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 7ed8fa7c7..f7d076a4a 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -57,32 +57,37 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
 
+      - name: Install uv
+        uses: astral-sh/setup-uv@v5
+        with:
+          enable-cache: true
+
       - name: Install dependencies
-        run: pip install '.[dev]'
+        run: uv sync --python ${{ matrix.python-version }} --extra dev
 
       - name: Run pytest
         run: |
           set -eEx
           set -o pipefail
 
-          python -m pytest -vv --cov
+          uv run python -m pytest -vv --cov
           # <100% coverage might indicate that some tests are not doing what they should,
           # but realistically, it might too hard to reach, so lower it a bit
           # (at the moment of writing, it is 99.95%)
-          coverage report --include='tests/*' --precision=2 --fail-under=97.00
+          uv run coverage report --include='tests/*' --precision=2 --fail-under=97.00
 
-          python -m pytest --dead-fixtures
+          uv run python -m pytest --dead-fixtures
 
-          python -m pytest -vv -m ci_only
+          uv run python -m pytest -vv -m ci_only
 
       - name: Test local installation
         run: |
           set -eE
           set -o pipefail
 
-          pip install .
-          pip uninstall -y cloudai
-          pip install -e .
+          uv pip install --system .
+          uv pip uninstall --system cloudai
+          uv pip install --system -e .
 
       - name: Test commands
         run: |

From 5a6db2dc8bf9ab5f72486a2c1a5d774584309ffa Mon Sep 17 00:00:00 2001
From: Ivan Podkidyshev <ipodkidyshev@nvidia.com>
Date: Mon, 15 Jun 2026 16:07:07 +0200
Subject: [PATCH 2/2] explicitly use python version

---
 .github/workflows/ci.yml | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index f7d076a4a..6b7bb7add 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -70,15 +70,15 @@ jobs:
           set -eEx
           set -o pipefail
 
-          uv run python -m pytest -vv --cov
+          uv run --python ${{ matrix.python-version }} --extra dev python -m pytest -vv --cov
           # <100% coverage might indicate that some tests are not doing what they should,
           # but realistically, it might too hard to reach, so lower it a bit
           # (at the moment of writing, it is 99.95%)
-          uv run coverage report --include='tests/*' --precision=2 --fail-under=97.00
+          uv run --python ${{ matrix.python-version }} --extra dev coverage report --include='tests/*' --precision=2 --fail-under=97.00
 
-          uv run python -m pytest --dead-fixtures
+          uv run --python ${{ matrix.python-version }} --extra dev python -m pytest --dead-fixtures
 
-          uv run python -m pytest -vv -m ci_only
+          uv run --python ${{ matrix.python-version }} --extra dev python -m pytest -vv -m ci_only
 
       - name: Test local installation
         run: |