Skip to content

Commit bb4937a

Browse files
committed
test(conversion): mock config loading in verdict tests
Assisted-by: OpenAI Codex Signed-off-by: Yuchen Wang <93700456+yuchenwang3@users.noreply.github.com>
1 parent 38fd978 commit bb4937a

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

‎tests/unit_tests/test_compare_mask_handling.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -166,6 +166,7 @@ def test_forward_comparison_verdict(self, monkeypatch, fail_on_mismatch, hf_valu
166166
monkeypatch.setattr(torch.Tensor, "cuda", lambda tensor: tensor)
167167
monkeypatch.setattr(torch.distributed, "is_initialized", lambda: False)
168168
monkeypatch.setattr(compare, "is_vision_language_model", lambda *args: False)
169+
monkeypatch.setattr(compare.AutoConfig, "from_pretrained", lambda *args, **kwargs: SimpleNamespace())
169170
monkeypatch.setattr(compare, "_load_megatron_model", lambda args: ([MagicMock()], MagicMock()))
170171
monkeypatch.setattr(compare, "_load_hf_model", lambda *args: MagicMock())
171172
monkeypatch.setattr(compare, "_setup_tokenizer_and_processor", lambda *args: (tokenizer, None))

0 commit comments

Comments
 (0)