From 0cbfe08a41a5f517a27fd432d4770feb7bef37af Mon Sep 17 00:00:00 2001
From: Luca Medeiros <67411094+luca-medeiros@users.noreply.github.com>
Date: Wed, 27 Jul 2022 01:04:36 +0900
Subject: [PATCH] Add lightning apps to info_packages of collect_env_details
 (#13815)

---
 requirements/collect_env_details.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/requirements/collect_env_details.py b/requirements/collect_env_details.py
index a39adf88cb..1d65753a55 100644
--- a/requirements/collect_env_details.py
+++ b/requirements/collect_env_details.py
@@ -27,6 +27,15 @@ import tqdm
 sys.path += [os.path.abspath(".."), os.path.abspath("")]
 import pytorch_lightning  # noqa: E402
 
+try:
+    import lightning
+except ModuleNotFoundError:
+    pass
+try:
+    import lightning_app
+except ModuleNotFoundError:
+    pass
+
 LEVEL_OFFSET = "\t"
 KEY_PADDING = 20
 
@@ -56,6 +65,8 @@ def info_packages():
         "pyTorch_version": torch.__version__,
         "pyTorch_debug": torch.version.debug,
         "pytorch-lightning": pytorch_lightning.__version__,
+        "lightning": lightning.__version__ if "lightning" in sys.modules else None,
+        "lightning_app": lightning_app.__version__ if "lightning_app" in sys.modules else None,
         "tqdm": tqdm.__version__,
     }