diff --git a/pkgs/development/python-modules/pylance/default.nix b/pkgs/development/python-modules/pylance/default.nix index 417c7b733c73..598b87e2906a 100644 --- a/pkgs/development/python-modules/pylance/default.nix +++ b/pkgs/development/python-modules/pylance/default.nix @@ -105,18 +105,6 @@ 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 [ - # AttributeError: module 'torch.distributed' has no attribute 'is_initialized' - "test_blob_api" - "test_convert_int_tensors" - "test_filtered_sampling_odd_batch_size" - "test_ground_truth" - "test_index_cast_centroids" - "test_index_with_no_centroid_movement" - "test_iter_filter" - "test_iter_over_dataset_fixed_shape_tensor" - "test_iter_over_dataset_fixed_size_lists" ]; meta = { diff --git a/pkgs/development/python-modules/pytorch-metric-learning/default.nix b/pkgs/development/python-modules/pytorch-metric-learning/default.nix index af57233b9cf2..de7ed5ebecd2 100644 --- a/pkgs/development/python-modules/pytorch-metric-learning/default.nix +++ b/pkgs/development/python-modules/pytorch-metric-learning/default.nix @@ -85,10 +85,6 @@ buildPythonPackage rec { "test_list_of_text" "test_untrained_indexer" ] - ++ lib.optionals stdenv.hostPlatform.isDarwin [ - # AttributeError: module 'torch.distributed' has no attribute 'init_process_group' - "test_single_proc" - ] ++ lib.optionals cudaSupport [ # crashes with SIGBART "test_accuracy_calculator_and_faiss_with_torch_and_numpy" diff --git a/pkgs/development/python-modules/skorch/default.nix b/pkgs/development/python-modules/skorch/default.nix index 76b9b7ef2994..e7053c8e6a2b 100644 --- a/pkgs/development/python-modules/skorch/default.nix +++ b/pkgs/development/python-modules/skorch/default.nix @@ -72,11 +72,15 @@ buildPythonPackage rec { # tries to import `transformers` and download HuggingFace data "skorch/tests/test_hf.py" ] - ++ lib.optionals (stdenv.hostPlatform.system != "x86_64-linux") [ - # torch.distributed is disabled by default in darwin - # aarch64-linux also failed these tests - "skorch/tests/test_history.py" - ]; + ++ lib.optionals + (stdenv.hostPlatform.system != "x86_64-linux" && stdenv.hostPlatform.system != "aarch64-darwin") + [ + # these tests fail when running in parallel for multiple platforms with: + # "RuntimeError: The server socket has failed to listen on any local + # network address because they use the same hardcoded port." For now, + # running for one platform per OS to avoid spurious failures. + "skorch/tests/test_history.py" + ]; pythonImportsCheck = [ "skorch" ];