lightning/tests/tests_fabric/utilities/test_runif.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

15 lines
250 B
Python
Raw Normal View History

2023-04-28 15:32:45 +00:00
import pytest
2023-05-04 23:21:58 +00:00
from tests_fabric.helpers.runif import RunIf
2023-04-28 15:32:45 +00:00
@RunIf(min_torch="99")
def test_always_skip():
exit(1)
@pytest.mark.parametrize("arg1", [0.5, 1.0, 2.0])
@RunIf(min_torch="0.0")
def test_wrapper(arg1: float):
assert arg1 > 0.0