Skip to content

Commit 5965241

Browse files
sgonorovsgonorov
committed
UV pipeline updates
1 parent ff2c99f commit 5965241

File tree

22 files changed

+115
-156
lines changed

22 files changed

+115
-156
lines changed

‎.github/actions/setup-uv/action.yml‎

Lines changed: 0 additions & 30 deletions
This file was deleted.

‎.github/actions/uv-sync/action.yml‎

Lines changed: 0 additions & 61 deletions
This file was deleted.

‎.github/workflows/linux.yml‎

Lines changed: 35 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -585,15 +585,19 @@ jobs:
585585

586586
- name: Setup UV
587587
if: ${{ matrix.test.run_condition }}
588-
uses: ./src/.github/actions/setup-uv
588+
uses: astral-sh/setup-uv@v4
589589

590590
- name: Install GenAI Wheels
591591
if: ${{ matrix.test.run_condition }}
592-
uses: ./src/.github/actions/uv-sync
592+
uses: ./src/.github/actions/install_wheel
593593
with:
594594
packages: "openvino;openvino_tokenizers[transformers];openvino_genai;whowhatbench"
595-
dependency-groups: "testing-full"
596-
local-wheel-dir: ${{ env.INSTALL_DIR }}/wheels
595+
local_wheel_dir: ${{ env.INSTALL_DIR }}/wheels
596+
597+
- name: Sync dependency groups
598+
if: ${{ matrix.test.run_condition }}
599+
working-directory: ${{ github.workspace }}/src
600+
run: uv sync --extra testing-full --no-install-project --prerelease=allow
597601

598602
- name: Tests
599603
if: ${{ matrix.test.run_condition }}
@@ -696,15 +700,21 @@ jobs:
696700

697701
- name: Setup UV
698702
if: ${{ matrix.test.run_condition }}
699-
uses: ./src/.github/actions/setup-uv
703+
uses: astral-sh/setup-uv@v4
700704

701705
- name: Install GenAI wheels
702706
if: ${{ matrix.test.run_condition }}
703-
uses: ./src/.github/actions/uv-sync
707+
uses: ./src/.github/actions/install_wheel
704708
with:
705709
packages: "openvino;openvino_tokenizers[transformers];openvino_genai[testing]"
706-
dependency-groups: "samples,testing-full"
707-
local-wheel-dir: ${{ env.INSTALL_DIR }}/wheels
710+
local_wheel_dir: ${{ env.INSTALL_DIR }}/wheels
711+
712+
- name: Sync dependency groups
713+
if: ${{ matrix.test.run_condition }}
714+
working-directory: ${{ github.workspace }}/src
715+
run: |
716+
uv sync --extra samples --no-install-project --prerelease=allow
717+
uv sync --extra testing-full --no-install-project --prerelease=allow
708718
709719
- name: Setup Node
710720
if: ${{ matrix.test.run_condition }}
@@ -785,15 +795,20 @@ jobs:
785795
run: chmod +x ${{ env.INSTALL_DIR }}/samples_bin/*
786796

787797
- name: Setup UV
788-
uses: ./src/.github/actions/setup-uv
798+
uses: astral-sh/setup-uv@v4
789799

790800
- name: Install GenAI wheels
791-
uses: ./src/.github/actions/uv-sync
801+
uses: ./src/.github/actions/install_wheel
792802
with:
793803
packages: "openvino;openvino_tokenizers[transformers];openvino_genai[testing]"
794-
dependency-groups: "samples,testing-full"
795-
workspace-member: "llm_bench"
796-
local-wheel-dir: ${{ env.INSTALL_DIR }}/wheels
804+
local_wheel_dir: ${{ env.INSTALL_DIR }}/wheels
805+
806+
- name: Sync dependency groups
807+
working-directory: ${{ github.workspace }}/src
808+
run: |
809+
uv sync --extra samples --no-install-project --prerelease=allow
810+
uv sync --extra testing-full --no-install-project --prerelease=allow
811+
uv sync --package llm_bench --prerelease=allow
797812
798813
- name: gtests unit tests
799814
if: ${{ fromJSON(needs.smart_ci.outputs.affected_components).continuous_batching }}
@@ -932,14 +947,17 @@ jobs:
932947
cmake --build ${{ env.BUILD_DIR}} --config Release --parallel $(nproc) --target py_openvino_genai --verbose
933948
934949
- name: Setup UV
935-
uses: ./src/.github/actions/setup-uv
950+
uses: astral-sh/setup-uv@v4
936951

937952
- name: Install dependencies
938-
uses: ./src/.github/actions/uv-sync
953+
uses: ./src/.github/actions/install_wheel
939954
with:
940955
packages: "openvino;openvino_tokenizers[transformers]"
941-
dependency-groups: "testing-full"
942-
local-wheel-dir: ${{ env.INSTALL_DIR }}/wheels
956+
local_wheel_dir: ${{ env.INSTALL_DIR }}/wheels
957+
958+
- name: Sync dependency groups
959+
working-directory: ${{ github.workspace }}/src
960+
run: uv sync --extra testing-full --no-install-project --prerelease=allow
943961

944962
- name: Run test_llm_pipeline.py when -DENABLE_XGRAMMAR=OFF
945963
env:

‎.github/workflows/mac.yml‎

Lines changed: 27 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -512,15 +512,19 @@ jobs:
512512

513513
- name: Setup UV
514514
if: ${{ matrix.test.run_condition }}
515-
uses: ./src/.github/actions/setup-uv
515+
uses: astral-sh/setup-uv@v4
516516

517517
- name: Install GenAI Wheels
518518
if: ${{ matrix.test.run_condition }}
519-
uses: ./src/.github/actions/uv-sync
519+
uses: ./src/.github/actions/install_wheel
520520
with:
521521
packages: "openvino;openvino_tokenizers[transformers];openvino_genai;whowhatbench"
522-
dependency-groups: "testing-full"
523-
local-wheel-dir: ${{ env.INSTALL_DIR }}/wheels
522+
local_wheel_dir: ${{ env.INSTALL_DIR }}/wheels
523+
524+
- name: Sync dependency groups
525+
if: ${{ matrix.test.run_condition }}
526+
working-directory: ${{ github.workspace }}/src
527+
run: uv sync --extra testing-full --no-install-project --prerelease=allow
524528

525529
- name: Tests
526530
if: ${{ matrix.test.run_condition }}
@@ -598,15 +602,21 @@ jobs:
598602

599603
- name: Setup UV
600604
if: ${{ matrix.test.run_condition }}
601-
uses: ./src/.github/actions/setup-uv
605+
uses: astral-sh/setup-uv@v4
602606

603607
- name: Install GenAI wheels
604608
if: ${{ matrix.test.run_condition }}
605-
uses: ./src/.github/actions/uv-sync
609+
uses: ./src/.github/actions/install_wheel
606610
with:
607611
packages: "openvino;openvino_tokenizers[transformers];openvino_genai[testing]"
608-
dependency-groups: "samples,testing-full"
609-
local-wheel-dir: ${{ env.INSTALL_DIR }}/wheels
612+
local_wheel_dir: ${{ env.INSTALL_DIR }}/wheels
613+
614+
- name: Sync dependency groups
615+
if: ${{ matrix.test.run_condition }}
616+
working-directory: ${{ github.workspace }}/src
617+
run: |
618+
uv sync --extra samples --no-install-project --prerelease=allow
619+
uv sync --extra testing-full --no-install-project --prerelease=allow
610620
611621
# transformers >= 4.52 require torch >= 2.6 and raise an error otherwise:
612622
# ValueError: Due to a serious vulnerability issue in `torch.load`, even with `weights_only=True`, we now require users to upgrade torch to at least v2.6 in order to use the function. This version restriction does not apply when loading files with safetensors.
@@ -693,14 +703,19 @@ jobs:
693703
cache: 'pip'
694704

695705
- name: Setup UV
696-
uses: ./src/.github/actions/setup-uv
706+
uses: astral-sh/setup-uv@v4
697707

698708
- name: Install GenAI wheels
699-
uses: ./src/.github/actions/uv-sync
709+
uses: ./src/.github/actions/install_wheel
700710
with:
701711
packages: "openvino;openvino_tokenizers[transformers];openvino_genai[testing]"
702-
dependency-groups: "samples,testing-full"
703-
local-wheel-dir: ${{ env.INSTALL_DIR }}/wheels
712+
local_wheel_dir: ${{ env.INSTALL_DIR }}/wheels
713+
714+
- name: Sync dependency groups
715+
working-directory: ${{ github.workspace }}/src
716+
run: |
717+
uv sync --extra samples --no-install-project --prerelease=allow
718+
uv sync --extra testing-full --no-install-project --prerelease=allow
704719
705720
- name: gtests unit tests
706721
run: |

‎.github/workflows/manylinux_2_28.yml‎

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -523,15 +523,19 @@ jobs:
523523

524524
- name: Setup UV
525525
if: ${{ matrix.test.run_condition }}
526-
uses: ./src/.github/actions/setup-uv
526+
uses: astral-sh/setup-uv@v4
527527

528528
- name: Install GenAI Wheels
529529
if: ${{ matrix.test.run_condition }}
530-
uses: ./src/.github/actions/uv-sync
530+
uses: ./src/.github/actions/install_wheel
531531
with:
532532
packages: "openvino;openvino_tokenizers[transformers];openvino_genai;whowhatbench"
533-
dependency-groups: "testing-full"
534-
local-wheel-dir: ${{ env.INSTALL_DIR }}/wheels
533+
local_wheel_dir: ${{ env.INSTALL_DIR }}/wheels
534+
535+
- name: Sync dependency groups
536+
if: ${{ matrix.test.run_condition }}
537+
working-directory: ${{ github.workspace }}/src
538+
run: uv sync --extra testing-full --no-install-project --prerelease=allow
535539

536540
- name: Tests
537541
if: ${{ matrix.test.run_condition }}

‎.github/workflows/windows.yml‎

Lines changed: 30 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -674,15 +674,20 @@ jobs:
674674

675675
- name: Setup UV
676676
if: ${{ matrix.test.run_condition }}
677-
uses: ./src/.github/actions/setup-uv
677+
uses: astral-sh/setup-uv@v4
678678

679679
- name: Install GenAI Wheels
680680
if: ${{ matrix.test.run_condition }}
681-
uses: ./src/.github/actions/uv-sync
681+
uses: ./src/.github/actions/install_wheel
682682
with:
683683
packages: "openvino;openvino_tokenizers[transformers];openvino_genai;whowhatbench"
684-
dependency-groups: "testing-full"
685-
local-wheel-dir: ${{ env.INSTALL_DIR }}/wheels
684+
local_wheel_dir: ${{ env.INSTALL_DIR }}/wheels
685+
686+
- name: Sync dependency groups
687+
if: ${{ matrix.test.run_condition }}
688+
working-directory: ${{ github.workspace }}/src
689+
shell: bash
690+
run: uv sync --extra testing-full --no-install-project --prerelease=allow
686691

687692
- name: Tests
688693
if: ${{ matrix.test.run_condition }}
@@ -777,15 +782,22 @@ jobs:
777782

778783
- name: Setup UV
779784
if: ${{ matrix.test.run_condition }}
780-
uses: ./src/.github/actions/setup-uv
785+
uses: astral-sh/setup-uv@v4
781786

782787
- name: Install GenAI wheels
783788
if: ${{ matrix.test.run_condition }}
784-
uses: ./src/.github/actions/uv-sync
789+
uses: ./src/.github/actions/install_wheel
785790
with:
786791
packages: "openvino;openvino_tokenizers[transformers];openvino_genai[testing]"
787-
dependency-groups: "samples,testing-full"
788-
local-wheel-dir: ${{ env.INSTALL_DIR }}/wheels
792+
local_wheel_dir: ${{ env.INSTALL_DIR }}/wheels
793+
794+
- name: Sync dependency groups
795+
if: ${{ matrix.test.run_condition }}
796+
working-directory: ${{ github.workspace }}/src
797+
shell: bash
798+
run: |
799+
uv sync --extra samples --no-install-project --prerelease=allow
800+
uv sync --extra testing-full --no-install-project --prerelease=allow
789801
790802
- name: Setup NodeJS
791803
if: ${{ matrix.test.run_condition }}
@@ -859,14 +871,20 @@ jobs:
859871
cache: 'pip'
860872

861873
- name: Setup UV
862-
uses: ./src/.github/actions/setup-uv
874+
uses: astral-sh/setup-uv@v4
863875

864876
- name: Install GenAI wheels
865-
uses: ./src/.github/actions/uv-sync
877+
uses: ./src/.github/actions/install_wheel
866878
with:
867879
packages: "openvino;openvino_tokenizers[transformers];openvino_genai[testing]"
868-
dependency-groups: "samples,testing-full"
869-
local-wheel-dir: ${{ env.INSTALL_DIR }}/wheels
880+
local_wheel_dir: ${{ env.INSTALL_DIR }}/wheels
881+
882+
- name: Sync dependency groups
883+
working-directory: ${{ github.workspace }}/src
884+
shell: bash
885+
run: |
886+
uv sync --extra samples --no-install-project --prerelease=allow
887+
uv sync --extra testing-full --no-install-project --prerelease=allow
870888
871889
- name: gtests unit tests
872890
run: |

‎README.md‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ Continuous batching functionality is used within OpenVINO Model Server (OVMS) to
7777
uv pip install openvino-genai
7878

7979
# With sample dependencies
80-
uv sync --group samples --group samples-export
80+
uv sync --extra samples --extra samples-export
8181
```
8282

8383
<a id="text-to-text"></a>

0 commit comments

Comments
 (0)