fix: Use plain int python type for num_pixels

Signed-off-by: Nikos Livathinos <nli@zurich.ibm.com>

Author: nikos-livathinos

docling_eval/evaluators/pixel/confusion_matrix_exporter.py

@@ -107,7 +107,7 @@ def __init__(
     def build_ds_report(
         self,
         num_images: int,
-        num_pixels: np.uint64,
+        num_pixels: int,
         headers: list[str],
         matrix_evaluation: MultiLabelMatrixEvaluation,
         collapsed_headers: list[str],

docling_eval/evaluators/pixel/pixel_types.py

@@ -75,7 +75,7 @@ class PagePixelLayoutEvaluation(BaseModel):
 class DatasetPixelLayoutEvaluation(BaseModel):
     layout_model_name: Optional[str]
     num_pages: int
-    num_pixels: np.uint64
+    num_pixels: int
     rejected_samples: Dict[EvaluationRejectionType, int]
     matrix_evaluation: MultiLabelMatrixEvaluation
     page_evaluations: Dict[str, PagePixelLayoutEvaluation]

docling_eval/evaluators/pixel_layout_evaluator.py

@@ -196,7 +196,7 @@ def __call__(
         all_pages_evaluations: Dict[str, PagePixelLayoutEvaluation] = (
             {}
         )  # Key is doc_id-page-no
-        ds_num_pixels: np.uint64 = np.uint64(0)
+        ds_num_pixels = 0
         self._layout_model_name = None
         pages_detailed_f1: list[float] = (
             []
@@ -385,7 +385,7 @@ def _compute_document_confusion_matrix(
         pred_doc: DoclingDocument,
     ) -> Tuple[
         Dict[int, np.ndarray],  # page_no -> page confusion matrix
-        np.uint64,  # num_pixels
+        int,  # num_pixels
     ]:
         r"""
         Compute the confusion matrix for the given documents.
@@ -405,7 +405,7 @@ def _compute_document_confusion_matrix(
         num_categories = len(matrix_categories_ids)
         off_diagonal_cells = num_categories * num_categories - num_categories
         page_confusion_matrices: Dict[int, np.ndarray] = {}
-        num_pixels: np.uint64 = np.uint64(0)
+        num_pixels = 0
 
         for page_no in sorted(gt_pages):
             page_size = true_doc.pages[page_no].size