suppres warnings (#617)

* suppres warnings * chlog
Lightning-Universe · Apr 8, 2021 · 35f6314 · 35f6314
1 parent b97c846
commit 35f6314
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -5,6 +5,13 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
 
+## [0.3.3] - 2021-04-DD
+
+### Changed
+
+- Suppressed missing package warnings, conditioned by `WARN_MISSING_PACKAGE="1"`  ([#617](https://github.com/PyTorchLightning/lightning-bolts/pull/617))
+
+
 ## [0.3.2] - 2021-03-20
 
 ### Changed

diff --git a/pl_bolts/utils/warnings.py b/pl_bolts/utils/warnings.py
@@ -4,6 +4,8 @@
 
 MISSING_PACKAGE_WARNINGS: Dict[str, int] = {}
 
+WARN_MISSING_PACKAGE = int(os.environ.get('WARN_MISSING_PACKAGE', False))
+
 
 def warn_missing_pkg(
     pkg_name: str,
@@ -23,6 +25,9 @@ def warn_missing_pkg(
     Returns:
         Number of warning calls
     """
+    if not WARN_MISSING_PACKAGE:
+        return -1
+
     if pkg_name not in MISSING_PACKAGE_WARNINGS:
         extra_text = os.linesep + extra_text if extra_text else ''
         if not pypi_name: