From 0b60aea1a711a29eac554dee552e1ceb113b1f6d Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Mon, 13 Jan 2025 13:18:25 +0100
Subject: [PATCH] python312Packages.mmcv: disable failing tests on
 aarch64-linux

---
 .../python-modules/mmcv/default.nix           | 28 ++++++++++++-------
 1 file changed, 18 insertions(+), 10 deletions(-)

diff --git a/pkgs/development/python-modules/mmcv/default.nix b/pkgs/development/python-modules/mmcv/default.nix
index c86ea25b50c0..3f7b5f9f4ce9 100644
--- a/pkgs/development/python-modules/mmcv/default.nix
+++ b/pkgs/development/python-modules/mmcv/default.nix
@@ -1,5 +1,6 @@
 {
   lib,
+  stdenv,
   buildPythonPackage,
   fetchFromGitHub,
 
@@ -120,16 +121,23 @@ buildPythonPackage rec {
   # test_cnn test_ops really requires gpus to be useful.
   # some of the tests take exceedingly long time.
   # the rest of the tests are disabled due to sandbox env.
-  disabledTests = [
-    "test_cnn"
-    "test_ops"
-    "test_fileclient"
-    "test_load_model_zoo"
-    "test_processing"
-    "test_checkpoint"
-    "test_hub"
-    "test_reader"
-  ];
+  disabledTests =
+    [
+      "test_cnn"
+      "test_ops"
+      "test_fileclient"
+      "test_load_model_zoo"
+      "test_processing"
+      "test_checkpoint"
+      "test_hub"
+      "test_reader"
+    ]
+    ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [
+      # flaky numerical tests (AssertionError)
+      "test_ycbcr2rgb"
+      "test_ycbcr2bgr"
+      "test_tensor2imgs"
+    ];
 
   meta = {
     description = "Foundational Library for Computer Vision Research";