Type annotation tweak for 3.5 compatibility

LeapBeyond · Nov 1, 2020 · 856f291 · 856f291
1 parent ef3b3c6
commit 856f291
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/scrubadub/scrubbers.py b/scrubadub/scrubbers.py
@@ -181,12 +181,11 @@ def clean_documents(self, documents: Union[Sequence[str], Dict[str, str]], **kwa
 
         filth_list = self._post_process_filth_list(filth_list)
 
-        clean_documents: Union[Dict[str, str], Sequence[str]]
         if isinstance(documents, list):
             clean_documents = [
                 self._replace_text(text=text, filth_list=filth_list, document_name=str(name), **kwargs)
                 for name, text in enumerate(documents)
-            ]
+            ]  # type: Union[Dict[str, str], Sequence[str]]
         elif isinstance(documents, dict):
             clean_documents = {
                 name: self._replace_text(text=text, filth_list=filth_list, document_name=name, **kwargs)