From 29e310807cd589d90aabeb8ecade8ea5d2d67d8f Mon Sep 17 00:00:00 2001 From: SeanNaren Date: Thu, 26 Nov 2020 17:07:48 +0000 Subject: [PATCH] Fix import order --- benchmarks/test_sharded_correctness.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/test_sharded_correctness.py b/benchmarks/test_sharded_correctness.py index 23027c430d..0f6d365880 100644 --- a/benchmarks/test_sharded_correctness.py +++ b/benchmarks/test_sharded_correctness.py @@ -9,7 +9,7 @@ from torch.utils.data.distributed import DistributedSampler from pytorch_lightning import Trainer, seed_everything from pytorch_lightning.plugins.sharded_plugin import DDPShardedPlugin -from pytorch_lightning.utilities import NATIVE_AMP_AVALAIBLE, FAIRSCALE_AVAILABLE +from pytorch_lightning.utilities import FAIRSCALE_AVAILABLE, NATIVE_AMP_AVALAIBLE from tests.base.boring_model import BoringModel, RandomDataset