From 07f1a3e8cfd19bed87bc869f1469f3f720b3eedb Mon Sep 17 00:00:00 2001 From: svlandeg Date: Sun, 15 Feb 2026 20:30:35 +0100 Subject: [PATCH 1/2] 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 From a991ed8e391659ff702d610e31f6e764929542d5 Mon Sep 17 00:00:00 2001 From: svlandeg Date: Sun, 15 Feb 2026 20:32:28 +0100 Subject: [PATCH 2/2] cleanup --- tests/test_engine.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/test_engine.py b/tests/test_engine.py index 6586dcf..784ffcb 100644 --- a/tests/test_engine.py +++ b/tests/test_engine.py @@ -31,7 +31,6 @@ class MockModel: def __init__(self, vocab_size=262): # 256 bytes + 6 special tokens self.vocab_size = vocab_size self.config = MockConfig() - self._device = "cpu" self._device = torch.device("cpu") def get_device(self):