Fix CI: update test to expect ValueError for missing tied keys

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
Signed-off-by: Justin Chu <justinchu@microsoft.com>

Author: justinchuby

src/mobius/_weight_utils_test.py

@@ -227,11 +227,13 @@ def test_both_present_no_change(self):
         assert sd["model.embed_tokens.weight"].data_ptr() == t1.data_ptr()
         assert sd["lm_head.weight"].data_ptr() == t2.data_ptr()
 
-    def test_neither_present_no_change(self):
-        """If neither present, no change."""
+    def test_neither_present_raises(self):
+        """Raise ValueError if neither key is found (catches key mismatches)."""
+        import pytest
+
         sd = {"other.weight": torch.randn(4)}
-        tie_word_embeddings(sd)
-        assert list(sd.keys()) == ["other.weight"]
+        with pytest.raises(ValueError, match=r"neither.*found"):
+            tie_word_embeddings(sd)
 
     def test_custom_keys(self):
         """Custom embed/head keys."""

src/mobius/models/qwen_vl_test.py

@@ -11,10 +11,10 @@
 
 from mobius._configs import ArchitectureConfig, VisionConfig
 from mobius.models.qwen_vl import (
-    Qwen25VLCausalLMModel,
-    Qwen25VLDecoderModel,
     Qwen3VL3ModelCausalLMModel,
     Qwen3VLDecoderModel,
+    Qwen25VLCausalLMModel,
+    Qwen25VLDecoderModel,
 )
 
 # Tiny config for weight preprocessing tests (no graph build needed)
@@ -122,7 +122,8 @@ class TestQwen3VL3ModelCausalLMModelTiedWeights:
 
     def test_lm_head_present_when_tied(self):
         config = dataclasses.replace(
-            _BASE_CONFIG, model_type="qwen3_vl",
+            _BASE_CONFIG,
+            model_type="qwen3_vl",
         )
         model = Qwen3VL3ModelCausalLMModel(config)
         sd = _fake_state_dict_qwen3vl()
@@ -132,7 +133,8 @@ def test_lm_head_present_when_tied(self):
 
     def test_lm_head_shares_data_ptr_with_embed(self):
         config = dataclasses.replace(
-            _BASE_CONFIG, model_type="qwen3_vl",
+            _BASE_CONFIG,
+            model_type="qwen3_vl",
         )
         model = Qwen3VL3ModelCausalLMModel(config)
         sd = _fake_state_dict_qwen3vl()
@@ -148,7 +150,8 @@ class TestQwen3VLDecoderModelTiedWeights:
 
     def test_lm_head_present_when_tied(self):
         config = dataclasses.replace(
-            _BASE_CONFIG, model_type="qwen3_vl",
+            _BASE_CONFIG,
+            model_type="qwen3_vl",
         )
         model = Qwen3VLDecoderModel(config)
         sd = _fake_state_dict_qwen3vl()
@@ -158,7 +161,8 @@ def test_lm_head_present_when_tied(self):
 
     def test_lm_head_shares_data_ptr_with_embed(self):
         config = dataclasses.replace(
-            _BASE_CONFIG, model_type="qwen3_vl",
+            _BASE_CONFIG,
+            model_type="qwen3_vl",
         )
         model = Qwen3VLDecoderModel(config)
         sd = _fake_state_dict_qwen3vl()