pytorch-lightning

1
import pytest
2

3
from tests_fabric.helpers.runif import RunIf
4

5

6
@RunIf(min_torch="99")
7
def test_always_skip():
8
    exit(1)
9

10

11
@pytest.mark.parametrize("arg1", [0.5, 1.0, 2.0])
12
@RunIf(min_torch="0.0")
13
def test_wrapper(arg1: float):
14
    assert arg1 > 0.0
15