From 17eed38750b23ba8fac5be088d09382c0cb5ef40 Mon Sep 17 00:00:00 2001 From: Deep Mehta Date: Sat, 24 Jan 2026 15:34:29 +0530 Subject: [PATCH] fix: move _torch_available before usage and use importlib.util.find_spec MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Fixes ruff F821 (undefined name) and F401 (unused import) errors. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude --- tests-unit/execution_test/test_cache_provider.py | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/tests-unit/execution_test/test_cache_provider.py b/tests-unit/execution_test/test_cache_provider.py index f5a6e2d03..bc321838d 100644 --- a/tests-unit/execution_test/test_cache_provider.py +++ b/tests-unit/execution_test/test_cache_provider.py @@ -1,8 +1,15 @@ """Tests for external cache provider API.""" +import importlib.util import pytest from typing import Optional + +def _torch_available() -> bool: + """Check if PyTorch is available.""" + return importlib.util.find_spec("torch") is not None + + from comfy_execution.cache_provider import ( CacheProvider, CacheContext, @@ -347,12 +354,3 @@ class MockCacheProvider(CacheProvider): def on_store(self, context: CacheContext, value: CacheValue) -> None: self.stores.append((context, value)) - - -def _torch_available() -> bool: - """Check if PyTorch is available.""" - try: - import torch - return True - except ImportError: - return False