Skip to content

Commit b84bbbb

Browse files
fix adapter test
Signed-off-by: AngeloDanducci <angelo.danducci.ii@ibm.com>
1 parent 50292c2 commit b84bbbb

1 file changed

Lines changed: 1 addition & 4 deletions

File tree

test/backends/test_huggingface_unit.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -592,6 +592,7 @@ async def test_intrinsic_logits_populated_when_option_set(stub_backend):
592592
backend.model_id = "stub-model"
593593

594594
adapter = _make_intrinsic_adapter_stub()
595+
backend._added_adapters = {adapter.qualified_name: adapter}
595596

596597
class _FakeChatCompletionResponse:
597598
class _Choice:
@@ -619,10 +620,6 @@ def transform(self, chunk, rewritten):
619620
return chunk
620621

621622
with (
622-
patch(
623-
"mellea.backends.huggingface.get_adapter_for_intrinsic",
624-
return_value=adapter,
625-
),
626623
patch(
627624
"mellea.backends.huggingface.granite_formatters.IntrinsicsRewriter",
628625
_FakeRewriter,

0 commit comments

Comments
 (0)