import from PL

2022-09-28 12:21:27 +02:00 · 2022-09-28 12:21:27 +02:00 · af2e16be2a
parent eb9c9b7f0d
commit af2e16be2a
1 changed files with 1 additions and 1 deletions
--- 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: