python3Packages.torch: Disable duplicate openmp inclusion (#424873)

pkgs/development/python-modules/compressai/default.nix

@@ -98,6 +98,9 @@ buildPythonPackage rec {
     # Cause pytest to hang on Darwin after the tests are done
     "tests/test_eval_model.py"
     "tests/test_train.py"
+
+    # fails in sandbox as it tries to launch a web browser (which fails due to missing `osascript`)
+    "tests/test_plot.py::test_plot[plotly-ms-ssim-rgb]"
   ];
 
   meta = {

pkgs/development/python-modules/gpytorch/default.nix

@@ -1,6 +1,5 @@
 {
   lib,
-  stdenv,
   buildPythonPackage,
   fetchFromGitHub,
   jaxtyping,
@@ -57,9 +56,6 @@ buildPythonPackage rec {
     "test_t_matmul_matrix"
   ];
 
-  # Testing is broken on Darwin due to a `scikit-learn` issue, see https://github.com/NixOS/nixpkgs/issues/423831
-  doCheck = !stdenv.hostPlatform.isDarwin;
-
   meta = with lib; {
     description = "Highly efficient and modular implementation of Gaussian Processes, with GPU acceleration";
     homepage = "https://gpytorch.ai";

pkgs/development/python-modules/optuna-dashboard/default.nix

@@ -58,6 +58,12 @@ buildPythonPackage rec {
     streamlit
   ];
 
+  disabledTests = lib.optionals stdenv.hostPlatform.isDarwin [
+    # AttributeError: module 'numpy' has no attribute 'float128' ==> not available on 64-bit Darwin
+    "test_infer_sortable"
+    "test_serialize_numpy_floating"
+  ];
+
   # Disable tests that use playwright (needs network)
   disabledTestPaths = [
     "e2e_tests/test_dashboard/test_usecases/test_preferential_optimization.py"

pkgs/development/python-modules/pylance/default.nix

@@ -129,6 +129,10 @@ buildPythonPackage rec {
   ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [
     # OSError: LanceError(IO): Resources exhausted: Failed to allocate additional 1245184 bytes for ExternalSorter[0]...
     "test_merge_insert_large"
+  ]
+  ++ lib.optionals stdenv.hostPlatform.isDarwin [
+    # Build hangs after all the tests are run due to a torch subprocess not exiting
+    "test_multiprocess_loading"
   ];
 
   meta = {

pkgs/development/python-modules/sentence-transformers/default.nix

@@ -1,6 +1,5 @@
 {
   lib,
-  stdenv,
   buildPythonPackage,
   fetchFromGitHub,
 
@@ -123,10 +122,6 @@ buildPythonPackage rec {
     "tests/test_pretrained_stsb.py"
     "tests/test_sentence_transformer.py"
     "tests/test_train_stsb.py"
-  ]
-  ++ lib.optionals stdenv.hostPlatform.isDarwin [
-    # Segfault
-    "tests/test_sparse_tensor.py"
   ];
 
   # Sentence-transformer needs a writable hf_home cache

pkgs/development/python-modules/sqlite-vec/default.nix

@@ -80,9 +80,5 @@ buildPythonPackage rec {
   meta = sqlite-vec-c.meta // {
     description = "Python bindings for sqlite-vec";
     maintainers = [ lib.maintainers.sarahec ];
-    badPlatforms = [
-      # segfaults during test
-      "x86_64-darwin"
-    ];
   };
 }

pkgs/development/python-modules/torch/source/default.nix

@@ -526,7 +526,9 @@ buildPythonPackage rec {
     blas
     blas.provider
   ]
-  ++ lib.optionals stdenv.cc.isClang [ llvmPackages.openmp ]
+  # Including openmp leads to two copies being used. This segfaults on ARM.
+  # https://github.com/pytorch/pytorch/issues/149201#issuecomment-2776842320
+  # ++ lib.optionals stdenv.cc.isClang [ llvmPackages.openmp ]
   ++ lib.optionals cudaSupport (
     with cudaPackages;
     [

pkgs/development/python-modules/torcheval/default.nix

@@ -136,6 +136,54 @@ buildPythonPackage {
     # -- tests/metrics/test_toolkit.py --
     "tests/metrics/test_toolkit.py::MetricToolkitTest::test_metric_sync"
     "tests/metrics/test_toolkit.py::MetricCollectionToolkitTest::test_metric_collection_sync"
+
+    # Cannot access local process over IPv6 (nodename nor servname provided) even with __darwinAllowLocalNetworking
+    # Will hang, or appear to hang, with an 5 minute (default) timeout per test
+    "tests/metrics/aggregation/test_auc.py"
+    "tests/metrics/aggregation/test_cat.py"
+    "tests/metrics/aggregation/test_max.py"
+    "tests/metrics/aggregation/test_mean.py"
+    "tests/metrics/aggregation/test_min.py"
+    "tests/metrics/aggregation/test_sum.py"
+    "tests/metrics/aggregation/test_throughput.py"
+    "tests/metrics/classification/test_accuracy.py"
+    "tests/metrics/classification/test_auprc.py"
+    "tests/metrics/classification/test_auroc.py"
+    "tests/metrics/classification/test_binned_auprc.py"
+    "tests/metrics/classification/test_binned_auroc.py"
+    "tests/metrics/classification/test_binned_precision_recall_curve.py"
+    "tests/metrics/classification/test_confusion_matrix.py"
+    "tests/metrics/classification/test_f1_score.py"
+    "tests/metrics/classification/test_normalized_entropy.py"
+    "tests/metrics/classification/test_precision_recall_curve.py"
+    "tests/metrics/classification/test_precision.py"
+    "tests/metrics/classification/test_recall_at_fixed_precision.py"
+    "tests/metrics/classification/test_recall.py"
+    "tests/metrics/functional/classification/test_auroc.py"
+    "tests/metrics/ranking/test_click_through_rate.py::TestClickThroughRate::test_ctr_with_valid_input"
+    "tests/metrics/ranking/test_hit_rate.py::TestHitRate::test_hitrate_with_valid_input"
+    "tests/metrics/ranking/test_reciprocal_rank.py::TestReciprocalRank::test_mrr_with_valid_input"
+    "tests/metrics/ranking/test_retrieval_precision.py::TestRetrievalPrecision::test_retrieval_precision_multiple_updates_1_query"
+    "tests/metrics/ranking/test_retrieval_precision.py::TestRetrievalPrecision::test_retrieval_precision_multiple_updates_n_queries_without_nan"
+    "tests/metrics/ranking/test_weighted_calibration.py::TestWeightedCalibration::test_weighted_calibration_with_valid_input"
+    "tests/metrics/regression/test_mean_squared_error.py"
+    "tests/metrics/regression/test_r2_score.py"
+    "tests/metrics/test_synclib.py::SynclibTest::test_gather_uneven_multidim"
+    "tests/metrics/test_synclib.py::SynclibTest::test_gather_uneven"
+    "tests/metrics/test_synclib.py::SynclibTest::test_numeric_sync_state"
+    "tests/metrics/test_synclib.py::SynclibTest::test_sync_list_length"
+    "tests/metrics/text/test_bleu.py::TestBleu::test_bleu_multiple_examples_per_update"
+    "tests/metrics/text/test_bleu.py::TestBleu::test_bleu_multiple_updates"
+    "tests/metrics/text/test_perplexity.py::TestPerplexity::test_perplexity_with_ignore_index"
+    "tests/metrics/text/test_perplexity.py::TestPerplexity::test_perplexity"
+    "tests/metrics/text/test_word_error_rate.py::TestWordErrorRate::test_word_error_rate_with_valid_input"
+    "tests/metrics/text/test_word_information_lost.py::TestWordInformationLost::test_word_information_lost"
+    "tests/metrics/text/test_word_information_preserved.py::TestWordInformationPreserved::test_word_information_preserved_with_valid_input"
+    "tests/metrics/window/test_auroc.py"
+    "tests/metrics/window/test_click_through_rate.py::TestClickThroughRate::test_ctr_with_valid_input"
+    "tests/metrics/window/test_mean_squared_error.py"
+    "tests/metrics/window/test_normalized_entropy.py::TestWindowedBinaryNormalizedEntropy::test_ne_with_valid_input"
+    "tests/metrics/window/test_weighted_calibration.py::TestWindowedWeightedCalibration::test_weighted_calibration_with_valid_input"
   ];
 
   meta = {