Unverified Commit cedae5a2 authored by Nick Cao's avatar Nick Cao Committed by GitHub
Browse files

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

parents eae7606a 85038ff6

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

+3 −0
Original line number Diff line number Diff line
@@ -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

+0 −4
Original line number Diff line number Diff line 
{

  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

+6 −0
Original line number Diff line number Diff line
@@ -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

+4 −0
Original line number Diff line number Diff line
@@ -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

+0 −5
Original line number Diff line number Diff line 
{

  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