release/0.7.0: Add range validation

jeroenrook · Jan 13, 2019 · 548f306 · 548f306
1 parent 4ab2087
commit 548f306
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/sklearn_porter/utils/Logging.py b/sklearn_porter/utils/Logging.py
@@ -11,6 +11,9 @@ class Logging(object):
     @staticmethod
     def get_logger(name, level=0):
         """Setup a logging instance"""
+        level = 0 if not isinstance(level, int) else level
+        level = 0 if level < 0 else level
+        level = 4 if level > 4 else level
         console = logging.StreamHandler()
         level = [logging.NOTSET, logging.ERROR, logging.WARN, logging.INFO,
                  logging.DEBUG][level]