Skip to content

Commit 689d72d

Browse files
fix test
1 parent f55f525 commit 689d72d

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/torch_compile/unit/v1/spec_decode/test_medusa.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -242,7 +242,7 @@ def fake_torch_compile(model, backend, options, dynamic):
242242

243243
assert output is compiled_sentinel
244244
assert captured["model"] is model
245-
assert captured["backend"] == "rbln"
245+
assert captured["backend"] is medusa_module.logged_rbln_backend
246246
assert captured["dynamic"] is False
247247

248248
options = cast(dict[str, Any], captured["options"])

0 commit comments

Comments
 (0)