Merge pull request #35 from topper-123/patch-1

Update none_missing decorator
engarde-dev · May 27, 2017 · 2ad77b0 · 2ad77b0
2 parents 7b0bcfd + 217d656
commit 2ad77b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/engarde/decorators.py b/engarde/decorators.py
@@ -5,13 +5,13 @@
 
 import engarde.checks as ck
 
-def none_missing():
+def none_missing(columns=None):
     """Asserts that no missing values (NaN) are found"""
     def decorate(func):
         @wraps(func)
         def wrapper(*args, **kwargs):
             result = func(*args, **kwargs)
-            ck.none_missing(result)
+            ck.none_missing(result, columns=columns)
             return result
         return wrapper
     return decorate