.

Author: Borda

docs/source/common/trainer.rst

@@ -1121,7 +1121,7 @@ To define your own behavior, subclass the relevant class and pass it in. Here's
 
 .. code-block:: python
 
-    from pytorch_lightning.environments import cluster_environment
+    from pytorch_lightning.plugins.environments import cluster_environment
 
     class MyCluster(ClusterEnvironment):
 

docs/source/extensions/accelerators.rst

@@ -70,7 +70,7 @@ First, implement your own ClusterEnvironment. Here is the torch elastic implemen
     import os
     from pytorch_lightning import _logger as log
     from pytorch_lightning.utilities import rank_zero_warn
-    from pytorch_lightning.environments.cluster_environment import ClusterEnvironment
+    from pytorch_lightning.plugins.environments.cluster_environment import ClusterEnvironment
 
     class TorchElasticEnvironment(ClusterEnvironment):
 

tests/accelerators/legacy/test_accelerator_connector.py

@@ -20,7 +20,7 @@
 from pytorch_lightning import accelerators, Trainer
 from pytorch_lightning.accelerators import Accelerator
 from pytorch_lightning.callbacks import Callback
-from pytorch_lightning.environments import ClusterEnvironment, SLURMEnvironment, TorchElasticEnvironment
+from pytorch_lightning.plugins.environments import ClusterEnvironment, SLURMEnvironment, TorchElasticEnvironment
 from pytorch_lightning.utilities import DistributedType
 from tests.base.boring_model import BoringModel