From 07f1a3e8cfd19bed87bc869f1469f3f720b3eedb Mon Sep 17 00:00:00 2001
From: svlandeg <svlandeg@github.com>
Date: Sun, 15 Feb 2026 20:30:35 +0100
Subject: [PATCH] fix MockModel's device definition

---
 tests/test_engine.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test_engine.py b/tests/test_engine.py
index 0159111..6586dcf 100644
--- a/tests/test_engine.py
+++ b/tests/test_engine.py
@@ -32,6 +32,7 @@ class MockModel:
         self.vocab_size = vocab_size
         self.config = MockConfig()
         self._device = "cpu"
+        self._device = torch.device("cpu")
 
     def get_device(self):
         return self._device