From 8aec0d8f46306512bc69ccaf5cf254ab9440c848 Mon Sep 17 00:00:00 2001
From: asagi4 <130366179+asagi4@users.noreply.github.com>
Date: Thu, 18 Dec 2025 23:50:58 +0200
Subject: [PATCH] I can't run tests if this exists...

---
 prompt_control/test_encode.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/prompt_control/test_encode.py b/prompt_control/test_encode.py
index f1fa92a..0aa70ea 100644
--- a/prompt_control/test_encode.py
+++ b/prompt_control/test_encode.py
@@ -20,7 +20,6 @@ def run(f, *args):
         return getattr(f, f.FUNCTION)(*args)
 
 
-@mock.patch("torch.cuda.current_device", lambda: "cpu")
 class TestEncode(unittest.TestCase):
     @classmethod
     def setUpClass(cls):