diff --git a/pkgs/development/python-modules/pytorch-metric-learning/default.nix b/pkgs/development/python-modules/pytorch-metric-learning/default.nix index 1e689e131679..d019e2b62917 100644 --- a/pkgs/development/python-modules/pytorch-metric-learning/default.nix +++ b/pkgs/development/python-modules/pytorch-metric-learning/default.nix @@ -1,4 +1,5 @@ -{ lib +{ stdenv +, lib , buildPythonPackage , fetchFromGitHub , isPy27 @@ -56,6 +57,10 @@ buildPythonPackage rec { "test_pca" # flaky "test_distributed_classifier_loss_and_miner" + ] ++ lib.optionals (stdenv.isLinux && stdenv.isAarch64) [ + # RuntimeError: DataLoader worker (pid(s) <...>) exited unexpectedly + "test_global_embedding_space_tester" + "test_with_same_parent_label_tester" ]; meta = {