fix: update normalize_weight function after Pint upgrade

robotoff/prediction/ocr/product_weight.py

@@ -1,4 +1,5 @@
 import functools
+import math
 import re
 from typing import Optional, Union
 
@@ -42,7 +43,13 @@ def normalize_weight(value: str, unit: str) -> tuple[float, str]:
     else:
         raise ValueError(f"unknown unit: {quantity.u}")
 
-    return normalized_quantity.magnitude, normalized_unit
+    # Rounding errors due to float may occur with Pint,
+    # round normalized value to floor if there is no significant difference
+    normalized_value = normalized_quantity.magnitude
+    if math.isclose(math.floor(normalized_value), normalized_value):
+        normalized_value = math.floor(normalized_value)
+
+    return normalized_value, normalized_unit
 
 
 def is_valid_weight(weight_value: str) -> bool:

tests/unit/prediction/ocr/test_product_weight.py

@@ -1,3 +1,5 @@
+import math
+
 import pytest
 
 from robotoff.prediction.ocr.dataclass import OCRRegex
@@ -66,8 +68,9 @@ def test_product_weight_with_ending_mention_regex(input_str: str, is_match: bool
     ],
 )
 def test_normalize_weight(value: str, unit: str, expected: tuple[float, str]):
-    result = normalize_weight(value, unit)
-    assert result == expected
+    normalized_value, normalized_unit = normalize_weight(value, unit)
+    assert math.isclose(normalized_value, expected[0])
+    assert normalized_unit == expected[1]
 
 
 @pytest.mark.parametrize(