diff --git a/docs/source/clouds/cluster.rst b/docs/source/clouds/cluster.rst index ce594c0b5ea2a..414c58baa6bf6 100644 --- a/docs/source/clouds/cluster.rst +++ b/docs/source/clouds/cluster.rst @@ -214,7 +214,7 @@ If auto-resubmit is not desired, it can be turned off in the :class:`~pytorch_li .. code-block:: python - from pytorch_lightning.plugins import SLURMEnvironment + from pytorch_lightning.plugins.environments import SLURMEnvironment trainer = Trainer(plugins=[SLURMEnvironment(auto_requeue=False)])