From f7c15f256ceb7073f004d592e77dace98dcf2423 Mon Sep 17 00:00:00 2001 From: Matteo Bortolon Date: Wed, 1 Dec 2021 02:24:06 +0100 Subject: [PATCH] Correct SLURM wrong import (#10842) --- docs/source/clouds/cluster.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)])