From 1cf77157a17475a74e1c70c2d61cf7ad8053c92f Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Thu, 18 Sep 2025 13:56:17 -0700
Subject: [PATCH] Enable inference tests in CI

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

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 8851c48b7..cd31261a7 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -36,16 +36,16 @@ jobs:
           export UV_BREAK_SYSTEM_PACKAGES=true
           export UV_SYSTEM_PYTHON=true
           uv pip freeze | grep numpy > numpy_override.txt
-          uv pip install ".[dev]" --inexact --override numpy_override.txt
-      - name: Run tests
-        run: |
-          nvidia-smi
-          export OTEL_METRICS_EXPORTER=none
-          pytest -v tests/unit
+          uv pip install --torch-backend=auto ".[dev]" --inexact --override numpy_override.txt
       - name: Lint for errors
         run: |
-          pylint --rcfile=.pylintrc comfy/
-          pylint --rcfile=.pylintrc comfy_extras/
+          pylint --rcfile=.pylintrc comfy/ comfy_extras/ comfy_api/ comfy_api_nodes/
+      - name: Run unit tests
+        run: |
+          pytest -v tests/unit
+      - name: Run inference tests
+        run: |
+          pytest -v tests/inference
   build_and_execute_macos:
     environment: "Testing"
     name: Installation Test for macOS