From af2e16be2af71b885fabe5efed303e6cfc76bc0d Mon Sep 17 00:00:00 2001
From: otaj <ota@lightning.ai>
Date: Wed, 28 Sep 2022 12:21:27 +0200
Subject: [PATCH] import from PL

---
 src/lightning_app/components/training.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/lightning_app/components/training.py b/src/lightning_app/components/training.py
index b467615cca..78790f10a3 100644
--- a/src/lightning_app/components/training.py
+++ b/src/lightning_app/components/training.py
@@ -97,7 +97,7 @@ class PyTorchLightningScriptRunner(TracerPythonScript):
         if self.node_rank != 0:
             return {}, args, kwargs
 
-        from lightning.pytorch.serve import ServableModuleValidator
+        from pytorch_lightning.serve import ServableModuleValidator
 
         callbacks = kwargs.get("callbacks", [])
         if self.sanity_serving: