Hmm

Author: NicolasHug

test/test_encoders.py

@@ -1,6 +1,7 @@
 import io
 import json
 import os
+import platform
 import re
 import subprocess
 import sys
@@ -332,12 +333,15 @@ def test_against_cli(
                 assert_close = partial(assert_tensor_close_on_at_least, percentage=99)
         elif format == "wav":
             rtol, atol = 0, 1e-4
-        elif format == "mp3" and asset is SINE_MONO_S32 and num_channels == 2:
-            # Not sure why, this one needs slightly higher tol. With default
-            # tolerances, the check fails on ~1% of the samples, so that's
-            # probably fine. It might be that the FFmpeg CLI doesn't rely on
-            # libswresample for converting channels?
-            rtol, atol = 0, 1e-3
+        elif format == "mp3" and asset is SINE_MONO_S32:
+            if platform.machine.lower() == "aarch64":
+                rtol, atol = 0, 1e-2
+            elif num_channels == 2:
+                # Not sure why, this one needs slightly higher tol. With default
+                # tolerances, the check fails on ~1% of the samples, so that's
+                # probably fine. It might be that the FFmpeg CLI doesn't rely on
+                # libswresample for converting channels?
+                rtol, atol = 0, 1e-3
         else:
             rtol, atol = None, None
 

test/utils.py

@@ -139,7 +139,7 @@ def psnr(a, b, max_val=255) -> float:
 # not guarantee bit-for-bit equality across systems and architectures, so we
 # also cannot. We currently use Linux on x86_64 as our reference system.
 def assert_frames_equal(*args, **kwargs):
-    if sys.platform == "linux" and "x86" in platform.machine():
+    if sys.platform == "linux" and "x86" in platform.machine().lower():
         if args[0].device.type == "cuda":
             atol = 3 if cuda_version_used_for_building_torch() >= (13, 0) else 2
             if get_ffmpeg_major_version() == 4: