We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
test_torch_compile_compatibility
1 parent 8f07b0a commit 020d9efCopy full SHA for 020d9ef
tests/test_utils.py
@@ -1860,8 +1860,9 @@ def test_torch_compile_disable() -> None:
1860
assert model == compiled_model
1861
1862
1863
-def test_torch_compile_compatibility() -> None:
+def test_torch_compile_compatibility(caplog: pytest.LogCaptureFixture) -> None:
1864
"""Test if torch-compile compatibility is checked correctly."""
1865
from tiatoolbox.models.architecture.utils import is_torch_compile_compatible
1866
1867
- assert isinstance(is_torch_compile_compatible(), bool)
+ is_torch_compile_compatible()
1868
+ assert "torch.compile" in caplog.text
0 commit comments