Card

Git

Discussions

 1
 2
 3
 4
 5
 6
 7
 8
 9
10

import pytest

try:
    import torch.nn
except ImportError:
    torch = None

if torch is None:
    pytest.skip("torch not installed", allow_module_level=True)
pytest.importorskip("rich")