Skip test if pytorch isn't available

LLNL · Apr 17, 2024 · e04fefe · e04fefe
1 parent 6f06e37
commit e04fefe
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ci_test/unit_tests/test_unit_module_periodic_padding.py b/ci_test/unit_tests/test_unit_module_periodic_padding.py
@@ -6,6 +6,13 @@
 from torch import Tensor
 import torch.nn.functional as F
 
+try:
+    from torch import Tensor
+    import torch.nn.functional as F
+except:
+    pytest.skip("PyTorch is required to run this test.", allow_module_level=True)
+
+
 @test_util.lbann_test(check_gradients=False)
 def test_periodic_padding_2D():
     # Prepare reference output