fix(scripts): improve MSE logging in release tests

ssube · Nov 24, 2023 · 8df6ec2 · 8df6ec2
1 parent 34493e7
commit 8df6ec2
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/api/scripts/test-release.py b/api/scripts/test-release.py
@@ -476,11 +476,12 @@ def run_test(
         ref = Image.open(ref_name) if path.exists(ref_name) else None
 
         mse = find_mse(result, ref)
+        threshold = test.mse_threshold * mse_mult
 
-        if mse < (test.mse_threshold * mse_mult):
-            logger.info("MSE within threshold: %.5f < %.5f", mse, test.mse_threshold)
+        if mse < threshold:
+            logger.info("MSE within threshold: %.5f < %.5f", mse, threshold)
         else:
-            logger.warning("MSE above threshold: %.5f > %.5f", mse, test.mse_threshold)
+            logger.warning("MSE above threshold: %.5f > %.5f", mse, threshold)
             passed = False
 
     return passed