From 3548fcd3c215b92d5bc10a7144ee469d24219ad5 Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Wed, 11 Sep 2024 14:47:23 -0700
Subject: [PATCH] Tweak test

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

diff --git a/tests/unit/test_language_nodes.py b/tests/unit/test_language_nodes.py
index 340ecb43e..875810d6b 100644
--- a/tests/unit/test_language_nodes.py
+++ b/tests/unit/test_language_nodes.py
@@ -75,7 +75,9 @@ def mock_openai_client():
         yield instance
 
 
-def test_transformers_loader():
+def test_transformers_loader(has_gpu):
+    if not has_gpu:
+        pytest.skip("requires GPU")
     loader = TransformersLoader()
     model, = loader.execute("microsoft/Phi-3-mini-4k-instruct", "")
     assert isinstance(model, LanguageModel)