Tests fixed

Lightning-AI · Dec 17, 2021 · cca3840 · cca3840
1 parent aabb31a
commit cca3840
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pytorch_lightning/loggers/neptune.py b/pytorch_lightning/loggers/neptune.py
@@ -266,6 +266,11 @@ def __init__(
         prefix: str = "training",
         **neptune_run_kwargs,
     ):
+        if neptune is None:
+            raise ModuleNotFoundError(
+                "You want to use the `Neptune` logger which is not installed yet, install it with"
+                " `pip install neptune-client`."
+            )
 
         # verify if user passed proper init arguments
         self._verify_input_arguments(api_key, project, name, run, neptune_run_kwargs)