Merge branch 'main' into prep-release

Author: adamtheturtle

pyproject.toml

@@ -37,6 +37,10 @@ dynamic = [
 ]
 dependencies = [
     "flask",
+    # Pin numpy to avoid:
+    # https://github.com/pytorch/pytorch/issues/128860
+    # When we remove this, also remove the deptry ignore.
+    "numpy<2.0.0; sys_platform=='win32'",
     "pillow",
     "piq",
     "pydantic-settings",
@@ -305,10 +309,13 @@ pep621_dev_dependency_groups = [
 ]
 
 [tool.deptry.per_rule_ignores]
-# tzdata is needed on Windows for zoneinfo to work.
-# See https://docs.python.org/3/library/zoneinfo.html#data-sources.
 DEP002 = [
+    # tzdata is needed on Windows for zoneinfo to work.
+    # See https://docs.python.org/3/library/zoneinfo.html#data-sources.
     "tzdata",
+    # numpy is pinned to avoid:
+    # https://github.com/pytorch/pytorch/issues/128860.
+    "numpy",
 ]
 
 [tool.pyproject-fmt]

src/mock_vws/target_raters.py

@@ -30,7 +30,7 @@ def _get_brisque_target_tracking_rating(image_content: bytes) -> int:
     image_tensor = image_tensor.unsqueeze(0)
     try:
         brisque_score = piq.brisque(x=image_tensor, data_range=255)
-    except AssertionError:
+    except (AssertionError, IndexError):
         return 0
     return math.ceil(int(brisque_score.item()) / 20)