diff --git a/ci_test/unit_tests/test_unit_module_periodic_padding.py b/ci_test/unit_tests/test_unit_module_periodic_padding.py index 354b5b78f27..24f0b1c1fe1 100644 --- 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