Skip to content

Commit 020d9ef

Browse files
committed
🐛 Fix test_torch_compile_compatibility
1 parent 8f07b0a commit 020d9ef

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/test_utils.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1860,8 +1860,9 @@ def test_torch_compile_disable() -> None:
18601860
assert model == compiled_model
18611861

18621862

1863-
def test_torch_compile_compatibility() -> None:
1863+
def test_torch_compile_compatibility(caplog: pytest.LogCaptureFixture) -> None:
18641864
"""Test if torch-compile compatibility is checked correctly."""
18651865
from tiatoolbox.models.architecture.utils import is_torch_compile_compatible
18661866

1867-
assert isinstance(is_torch_compile_compatible(), bool)
1867+
is_torch_compile_compatible()
1868+
assert "torch.compile" in caplog.text

0 commit comments

Comments
 (0)