From ee653001dbcb6adc31f50570b183b36274877bbf Mon Sep 17 00:00:00 2001
From: gptbert <125624295+gptbert@users.noreply.github.com>
Date: Sun, 8 Oct 2023 19:34:22 +0800
Subject: [PATCH] Python 3.7 AttributeError: module 'math' has no attribute
 'prod'

Python 3.7 math.prod wat not exist, was new in version 3.8.

AttributeError: module 'math' has no attribute 'prod'

https://docs.python.org/3/library/math.html#math.prod
---
 nougat/utils/dataset.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/nougat/utils/dataset.py b/nougat/utils/dataset.py
index 59822eb..2cab589 100644
--- a/nougat/utils/dataset.py
+++ b/nougat/utils/dataset.py
@@ -7,8 +7,9 @@
 import logging
 import os
 from math import prod
+import operator
 from pathlib import Path
-from functools import partial
+from functools import partial, reduce
 import random
 from typing import Dict, Tuple, Callable
 from PIL import Image, UnidentifiedImageError
@@ -247,7 +248,13 @@ def __getitem__(self, idx: int) -> Tuple[torch.Tensor, torch.Tensor]:
         if sample is None:
             # if sample is broken choose another randomly
             return self[random.randint(0, self.dataset_length - 1)]
-        if sample is None or sample["image"] is None or prod(sample["image"].size) == 0:
+        try:
+            # Python 3.8+
+            is_empty_sample = prod(sample["image"].size) == 0
+        except:
+            # Python 3.7
+            is_empty_sample = reduce(operator.mul, sample["image"].size, 1) == 0
+        if sample is None or sample["image"] is None or is_empty_sample:
             input_tensor = None
         else:
             input_tensor = self.nougat_model.encoder.prepare_input(