type annotate validators

siddhantgoel · Aug 3, 2019 · ac3d485 · ac3d485
1 parent 973391a
commit ac3d485
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/streaming_form_data/validators.py b/streaming_form_data/validators.py
@@ -3,11 +3,11 @@ class ValidationError(Exception):
 
 
 class MaxSizeValidator:
-    def __init__(self, max_size):
+    def __init__(self, max_size: int):
         self.so_far = 0
         self.max_size = max_size
 
-    def __call__(self, chunk):
+    def __call__(self, chunk: bytes):
         self.so_far += len(chunk)
 
         if self.so_far > self.max_size: