From d6c374942e5a2e752c9f0831f5946d3a02ec683d Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Wed, 8 May 2024 17:03:45 -0700
Subject: [PATCH] Only run this test with CUDA. Fix CPU mode for testing later

---
 tests/unit/test_openapi_nodes.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/unit/test_openapi_nodes.py b/tests/unit/test_openapi_nodes.py
index 1ac001fbe..f141b69b8 100644
--- a/tests/unit/test_openapi_nodes.py
+++ b/tests/unit/test_openapi_nodes.py
@@ -11,7 +11,6 @@ from PIL import Image
 from freezegun import freeze_time
 
 from comfy.cmd import folder_paths
-from comfy.nodes.base_nodes import LoadImage
 from comfy_extras.nodes.nodes_open_api import SaveImagesResponse, IntRequestParameter, FloatRequestParameter, \
     StringRequestParameter, HashImage, StringPosixPathJoin, LegacyOutputURIs, DevNullUris, StringJoin, StringToUri, \
     UriFormat, ImageExifMerge, ImageExifCreationDateAndBatchNumber, ImageExif, ImageExifUncommon, \
@@ -257,6 +256,7 @@ def test_posix_join_curly_brackets():
     assert joined_path == "a_{test}/b/c"
 
 
+@pytest.mark.skipif(not torch.cuda.is_available(), reason="CUDA is not available")
 def test_file_request_parameter(use_temporary_input_directory):
     _image_1x1_px = np.array([[[255, 0, 0]]], dtype=np.uint8)
     image_path = os.path.join(use_temporary_input_directory, "test_image.png")
@@ -266,6 +266,7 @@ def test_file_request_parameter(use_temporary_input_directory):
     n = ImageRequestParameter()
     loaded_image, = n.execute(uri=image_path)
     assert loaded_image.shape == (1, 1, 1, 3)
+    from comfy.nodes.base_nodes import LoadImage
 
     load_image_node = LoadImage()
     load_image_node_rgb, _ = load_image_node.load_image(image=os.path.basename(image_path))