From cd360dcb26b08068c2ff29869df0ae6c5183891b Mon Sep 17 00:00:00 2001
From: Stas Bekman <stas00@users.noreply.github.com>
Date: Tue, 3 Nov 2020 10:17:11 -0800
Subject: [PATCH] [examples] minimal version requirement run-time check in PL
 (#8133)

Co-authored-by: Sam Shleifer <sshleifer@gmail.com>
---
 examples/lightning_base.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/examples/lightning_base.py b/examples/lightning_base.py
index 739e5dc596..3f35ffe0f0 100644
--- a/examples/lightning_base.py
+++ b/examples/lightning_base.py
@@ -7,6 +7,7 @@ from typing import Any, Dict
 import pytorch_lightning as pl
 from pytorch_lightning.utilities import rank_zero_info
 
+import pkg_resources
 from transformers import (
     AdamW,
     AutoConfig,
@@ -32,6 +33,15 @@ from transformers.optimization import (
 
 logger = logging.getLogger(__name__)
 
+try:
+    pkg = "pytorch_lightning"
+    min_ver = "1.0.4"
+    pkg_resources.require(f"{pkg}>={min_ver}")
+except pkg_resources.VersionConflict:
+    logger.warning(
+        f"{pkg}>={min_ver} is required for a normal functioning of this module, but found {pkg}=={pkg_resources.get_distribution(pkg).version}. Try pip install -r examples/requirements.txt"
+    )
+
 
 MODEL_MODES = {
     "base": AutoModel,