fix dash test runner

2026-07-11 21:59:43 +03:00
parent bd99c5177b
commit c195b5ce78
3 changed files with 34 additions and 5 deletions
--- a/tests/test_real_model_backend.py
+++ b/tests/test_real_model_backend.py
@@ -15,7 +15,7 @@ import pytest
 from meshnet_node.model_backend import (
    InsufficientVRAMError,
    PartialModelLoadUnsupported,
-    ShardCacheMiss,
+    KVCacheMiss,
    TensorPayload,
    TorchModelShard,
    _call_layer,
@@ -629,7 +629,7 @@ def test_shard_cache_decode_miss_is_explicit():
    torch = pytest.importorskip("torch")
    shard = _fake_cache_shard(torch)

-    with pytest.raises(ShardCacheMiss):
+    with pytest.raises(KVCacheMiss):
        shard._run_layers(
            torch.zeros((1, 1, 2), dtype=torch.bfloat16),
            torch.ones((1, 5), dtype=torch.long),