Add OTX dependency to Geti Tune app (#4977)

Author: leoll2

.github/workflows/backend-lint-and-test.yaml

@@ -74,7 +74,7 @@ jobs:
       - name: Install uv
         uses: astral-sh/setup-uv@3259c6206f993105e3a61b142c2d97bf4b9ef83d # v7.1.0
         with:
-          version: "0.8.8"
+          version: "0.9.7"
           enable-cache: false
 
       - name: Install OpenCV dependencies
@@ -88,7 +88,7 @@ jobs:
         working-directory: application/backend
         run: |
           uv lock --check
-          uv sync --frozen --all-extras
+          uv sync --frozen --extra cpu --extra mqtt
 
       - name: Check formatting with ruff
         working-directory: application/backend

.github/workflows/ui-lint-and-test.yaml

@@ -77,7 +77,7 @@ jobs:
       - name: Install uv
         uses: astral-sh/setup-uv@3259c6206f993105e3a61b142c2d97bf4b9ef83d # v7.1.0
         with:
-          version: "0.8.8"
+          version: "0.9.7"
           enable-cache: false
 
       - name: Install OpenCV dependencies
@@ -89,7 +89,7 @@ jobs:
         working-directory: application/backend
         run: |
           uv lock --check
-          uv sync --frozen --all-extras
+          uv sync --frozen --extra cpu
 
       - name: Get OpenAPI spec
         working-directory: application/backend
@@ -353,85 +353,6 @@ jobs:
           path: application/ui/playwright-report/
           retention-days: 30
 
-  # e2e-tests:
-  #   name: E2E Tests
-  #   needs:
-  #     - check_paths
-  #     - build
-  #     - generate-openapi-spec
-  #   if: needs.check_paths.outputs.run_workflow == 'true'
-  #   permissions:
-  #     contents: read
-  #   timeout-minutes: 30
-  #   runs-on: ubuntu-latest
-  #   container:
-  #     image: mcr.microsoft.com/playwright:v1.54.0-noble@sha256:18d6adb6aaccf1b0f30eba890069972e089138e4a59ddb5303d7e7290e4e38b6
-  #   steps:
-  #     - name: Checkout code
-  #       uses: actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8 # v5.0.0
-  #       with:
-  #         persist-credentials: false
-
-  #     - uses: actions/setup-node@a0853c24544627f65ddf259abe73b1d18a591444 # v5.0.0
-  #       with:
-  #         node-version-file: application/ui/.nvmrc
-
-  #     - name: Set up Python
-  #       uses: actions/setup-python@e797f83bcb11b83ae66e0230d6156d7c80228e7c # v6.0.0
-  #       with:
-  #         python-version: "3.13"
-
-  #     - name: Install uv
-  #       uses: astral-sh/setup-uv@d0cc045d04ccac9d8b7881df0226f9e82c39688e # v6.8.0
-  #       with:
-  #         version: "0.8.8"
-  #         enable-cache: false
-
-  #     - name: Install OpenCV dependencies
-  #       run: |
-  #         apt-get update
-  #         apt-get install -y libgl1 libglib2.0-0
-
-  #     - name: Setup Backend
-  #       working-directory: application/backend
-  #       run: |
-  #         uv sync --frozen --all-extras
-
-  #     - name: Download UI build
-  #       uses: actions/download-artifact@634f93cb2916e3fdff6788551b99b062d0335ce0 # v5.0.0
-  #       with:
-  #         name: ui-dist
-  #         path: application/ui
-
-  #     - name: Download OpenAPI spec
-  #       uses: actions/download-artifact@634f93cb2916e3fdff6788551b99b062d0335ce0 # v5.0.0
-  #       with:
-  #         name: openapi-spec
-  #         path: application/ui/src/api
-
-  #     - name: Unpack build
-  #       working-directory: "application/ui"
-  #       run: tar -xzf dist.tar.gz
-
-  #     - name: Install dependencies
-  #       working-directory: "application/ui"
-  #       run: npm ci
-
-  #     - name: Build OpenAPI type definitions
-  #       working-directory: "application/ui"
-  #       run: npm run build:api
-
-  #     - name: Run E2E tests
-  #       working-directory: application/ui
-  #       run: npm run test:e2e
-
-  #     - name: Upload E2E results
-  #       uses: actions/upload-artifact@ea165f8d65b6e75b540449e92b4886f43607fa02 # v4.6.2
-  #       if: always()
-  #       with:
-  #         name: e2e-test-results
-  #         path: application/ui/playwright-report/
-
   required_check:
     name: Required Check ui-lint-and-test
     needs:
@@ -441,7 +362,6 @@ jobs:
       - lint
       - unit-tests
       - playwright-tests
-      # - e2e-tests
     runs-on: ubuntu-latest
     env:
       CHECKS: ${{ join(needs.*.result, ' ') }}

application/backend/pyproject.toml

@@ -28,6 +28,14 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
+# Choose one of the following training backends:
+# - cpu: only CPU training (no GPU support)
+# - xpu: support training on compatible Intel GPUs
+# - cuda: support training on compatible NVIDIA GPUs
+cpu = ["otx[cpu]"]
+xpu = ["otx[xpu]"]
+cuda = ["otx[cuda]"]
+
 mqtt = [
     "paho-mqtt~=2.1.0",
 ]
@@ -42,19 +50,38 @@ dev = [
 ]
 lint = [
     "ruff==0.9.3",
-    "mypy~=1.17",
+    "mypy~=1.16.1",
     "types-PyYAML~=6.0.12",
     "types-requests~=2.32.4",
 ]
 
 [tool.uv]
-default-groups = "all"
-required-version = "~=0.8.8"
+default-groups = ["dev", "lint"]
+required-version = "~=0.9.7"
+conflicts = [
+    [
+        { extra = "cpu" },
+        { extra = "xpu" },
+        { extra = "cuda" },
+    ],
+    # redeclare library conflicts to workaround bug with uv dependency resolution
+    # For details, see: https://github.com/astral-sh/uv/issues/9942
+    [
+        { package = "otx", extra = "cpu" },
+        { package = "otx", extra = "xpu" },
+        { package = "otx", extra = "cuda" },
+    ],
+]
 
 [tool.uv.build-backend]
 module-root = ""
 module-name = "app"
 
+[tool.uv.sources]
+# TODO switch to local path after the 'feature/datumaro' branch is merged to develop
+# otx = { path = "../../library", editable = true }
+otx = { git = "https://github.com/open-edge-platform/training_extensions.git", branch = "feature/datumaro", subdirectory = "library" }
+
 [tool.ruff]
 target-version = "py313"
 line-length = 120