File tree Expand file tree Collapse file tree 3 files changed +9
-8
lines changed Expand file tree Collapse file tree 3 files changed +9
-8
lines changed Original file line number Diff line number Diff line change @@ -42,7 +42,6 @@ skip_glob = [
4242 " tests/loggers/*" ,
4343 " tests/metrics/*" ,
4444 " tests/models/*" ,
45- " tests/plugins/*" ,
4645 " tests/trainer/*" ,
4746 " tests/tuner/*" ,
4847]
Original file line number Diff line number Diff line change 1- from pytorch_lightning .callbacks import Callback
2- from pytorch_lightning .utilities import _APEX_AVAILABLE
3- from tests .base .boring_model import BoringModel
4- from pytorch_lightning import Trainer
5- import pytest
61import os
72from unittest import mock
3+
4+ import pytest
5+
6+ from pytorch_lightning import Trainer
7+ from pytorch_lightning .callbacks import Callback
88from pytorch_lightning .plugins .apex import ApexPlugin
9+ from pytorch_lightning .utilities import _APEX_AVAILABLE
10+ from tests .base .boring_model import BoringModel
911
1012
1113@pytest .mark .skipif (not _APEX_AVAILABLE , reason = "test requires apex" )
Original file line number Diff line number Diff line change 88from pytorch_lightning import Trainer
99from pytorch_lightning .callbacks import Callback
1010from pytorch_lightning .plugins .sharded_native_amp_plugin import ShardedNativeAMPPlugin
11- from pytorch_lightning .plugins .sharded_plugin import DDPShardedPlugin , _FAIRSCALE_AVAILABLE
12- from pytorch_lightning .utilities import _NATIVE_AMP_AVAILABLE , _APEX_AVAILABLE
11+ from pytorch_lightning .plugins .sharded_plugin import _FAIRSCALE_AVAILABLE , DDPShardedPlugin
12+ from pytorch_lightning .utilities import _APEX_AVAILABLE , _NATIVE_AMP_AVAILABLE
1313from pytorch_lightning .utilities .exceptions import MisconfigurationException
1414from tests .base .boring_model import BoringModel
1515
You can’t perform that action at this time.
0 commit comments