Correct object type hints

PUNCH-Cyber · Jul 24, 2019 · 6c0afae · 6c0afae
1 parent 6937cfe
commit 6c0afae
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/stoq/data_classes.py b/stoq/data_classes.py
@@ -17,7 +17,7 @@
 import uuid
 from copy import deepcopy
 from datetime import datetime
-from typing import Dict, List, Optional, DefaultDict
+from typing import Dict, List, Optional, DefaultDict, Union
 
 import stoq.helpers as helpers
 
@@ -57,7 +57,7 @@ def __repr__(self):
 class Payload:
     def __init__(
         self,
-        content: bytes,
+        content: Union[bytes, str],
         payload_meta: Optional[PayloadMeta] = None,
         extracted_by: Optional[str] = None,
         extracted_from: Optional[str] = None,
@@ -129,7 +129,7 @@ def __init__(
         size: int,
         payload_meta: PayloadMeta,
         workers: Dict[str, Dict],
-        plugins_run: Dict[str, List[List]],
+        plugins_run: Dict[str, List[str]],
         extracted_from: Optional[str] = None,
         extracted_by: Optional[str] = None,
     ) -> None: