missing import

analyze.py

@@ -1,3 +1,4 @@
+from itertools import count
 from typing import Any, Iterable, Literal, Optional, TypedDict, TypeVar, Union, overload
 
 from datasets import Features, Value, get_dataset_config_info
@@ -7,7 +8,7 @@ from presidio_analyzer import AnalyzerEngine, BatchAnalyzerEngine, RecognizerRes
 
 Row = dict[str, Any]
 T = TypeVar("T")
-BATCH_SIZE = 10
+BATCH_SIZE = 1
 batch_analyzer: Optional[BatchAnalyzerEngine] = None