fix(benchmark): Lint

Author: flooie

benchmark/benchmark.py

@@ -8,10 +8,11 @@
 import sys
 from io import StringIO
 from pathlib import Path
+from typing import Any, Dict
 
 from matplotlib import pyplot as plt  # type: ignore
 
-from eyecite import clean_text, get_citations
+from eyecite import get_citations
 
 SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
 sys.path.append(os.path.dirname(SCRIPT_DIR))
@@ -45,20 +46,22 @@ def generate_branch_report(self, branch: str) -> None:
         now = datetime.datetime.now()
         data = []
         for row in csv_data:
-            text = (
+            text: str = (
                 row["xml_harvard"]
                 or row["html_lawbox"]
                 or row["html_columbia"]
                 or row["html_anon_2020"]
                 or row["html"]
             )
-            params = {"clean_steps": ["html", "inline_whitespace"]}
+            params: Dict[str, Any] = {
+                "clean_steps": ["html", "inline_whitespace"]
+            }
             if text:
                 # Remove XML encodings from xml_harvard
                 text = re.sub(r"^<\?xml.*?\?>", "", text, count=1)
-                params['markup_text'] = text or ""
+                params["markup_text"] = text or ""
             else:
-                params['markup_text'] = row['plain_text']
+                params["markup_text"] = row["plain_text"]
 
             found_citations = get_citations(**params)