Fixed Import in Docs For Multinode Trainer Name Which does Not Exist (#…

…15663)
Lightning-AI · Nov 13, 2022 · 23f88cd · 23f88cd
1 parent 57ac548
commit 23f88cd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/source-app/levels/basic/hello_components/pl_multinode.py b/docs/source-app/levels/basic/hello_components/pl_multinode.py
@@ -1,6 +1,6 @@
 # app.py
 import lightning as L
-from lightning.app.components import PyTorchLightningMultiNode
+from lightning.app.components import LightningTrainerMultiNode
 from lightning.pytorch.demos.boring_classes import BoringModel
 
 
@@ -12,7 +12,7 @@ def run():
         trainer.fit(model)
 
 # 8 GPU: (2 nodes of 4 x v100)
-component = PyTorchLightningMultiNode(
+component = LightningTrainerMultiNode(
     LightningTrainerDistributed,
     num_nodes=2,
     cloud_compute=L.CloudCompute("gpu-fast-multi"), # 4 x v100