Skip to content

Commit e033f51

Browse files
committed
skip test for Windows
1 parent e2fafe5 commit e033f51

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/plugins/test_custom_plugin.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,21 @@
11
import pytest
22
import torch
3+
34
from pytorch_lightning import Trainer
45
from pytorch_lightning.plugins import DDPPlugin
56
from tests.helpers import BoringModel
7+
from tests.helpers.runif import RunIf
68

79

810
class CustomParallelPlugin(DDPPlugin):
11+
912
def __init__(self, **kwargs):
1013
super().__init__(**kwargs)
1114
# Set to None so it will be overwritten by the accelerator connector.
1215
self.sync_batchnorm = None
1316

1417

18+
@RunIf(skip_windows=True)
1519
@pytest.mark.skipif(torch.cuda.is_available(), reason="test doesn't requires GPU machine")
1620
def test_sync_batchnorm_set(tmpdir):
1721
"""Tests if sync_batchnorm is automatically set for custom plugin."""

0 commit comments

Comments
 (0)