From 1820fb19ea5d200e8cd2a09bcbbcc35926f13228 Mon Sep 17 00:00:00 2001
From: Vito Sansevero <vito.sansevero@gmail.com>
Date: Fri, 16 Oct 2026 17:29:08 -0700
Subject: [PATCH] test(ci): enforce reviewed CPU dependency versions

---
 .github/workflows/tests.yml          | 2 +-
 tests/comfyui/image-browser.spec.mjs | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index eacb923..d5db95b 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -93,7 +93,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install "torch==${{ steps.runtime.outputs.torchVersion }}" "torchvision==${{ steps.runtime.outputs.torchvisionVersion }}" --index-url https://download.pytorch.org/whl/cpu
-          pip install -r .ci-comfyui/requirements.txt -r requirements-dev.txt
+          pip install -r .ci-comfyui/requirements.txt -r requirements-dev.txt "torch==${{ steps.runtime.outputs.torchVersion }}" "torchvision==${{ steps.runtime.outputs.torchvisionVersion }}"
       - run: npm ci --prefix tools/testing --ignore-scripts
       - run: npx --prefix tools/testing playwright install --with-deps chromium
       - name: Classic and Nodes 2.0 startup and compatibility
diff --git a/tests/comfyui/image-browser.spec.mjs b/tests/comfyui/image-browser.spec.mjs
index ae7474e..dd258d6 100644
--- a/tests/comfyui/image-browser.spec.mjs
+++ b/tests/comfyui/image-browser.spec.mjs
@@ -10,6 +10,7 @@ const web = resolve(fileURLToPath(new URL('../../web/', import.meta.url)));
 test.beforeEach(async ({ page, context, request, baseURL }, testInfo) => {
     const info = await (await request.get(`${baseURL}system_stats`)).json();
     expect(info.system.comfyui_version).toBe(runtime.serverVersion);
+    if (process.env.KIKO_COMFY_ROOT) expect(info.system.pytorch_version).toBe(`${runtime.torchVersion}+cpu`);
     expect(info.system.required_frontend_version).toBe(runtime.frontendVersion);
     expect(info.system.comfy_package_versions.find(item => item.name === 'comfyui-frontend-package')?.installed).toBe(runtime.frontendVersion);
     await testInfo.attach('host-versions.json', { body: JSON.stringify(info.system, null, 2), contentType: 'application/json' });