From 545716083af3d1b486205cd1cd3143a56af3146b Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Mon, 25 Nov 2024 01:09:08 +0100
Subject: [PATCH] python312Packages.dask-ml: disable broken tests

---
 .../python-modules/dask-ml/default.nix           | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/pkgs/development/python-modules/dask-ml/default.nix b/pkgs/development/python-modules/dask-ml/default.nix
index 427f1b446c21..913480c060d3 100644
--- a/pkgs/development/python-modules/dask-ml/default.nix
+++ b/pkgs/development/python-modules/dask-ml/default.nix
@@ -75,6 +75,11 @@ buildPythonPackage rec {
       # AttributeError: 'csr_matrix' object has no attribute 'A'
       # Fixed in https://github.com/dask/dask-ml/pull/996
       "tests/test_svd.py"
+
+      # Tests fail with dask>=0.11.2
+      # RuntimeError: Not enough arguments provided
+      # Reported in https://github.com/dask/dask-ml/issues/1003
+      "tests/model_selection/test_incremental.py"
     ]
     ++ lib.optionals stdenv.isDarwin [
       # RuntimeError: Not enough arguments provided: missing keys
@@ -87,6 +92,17 @@ buildPythonPackage rec {
   disabledTests = [
     # Flaky: `Arrays are not almost equal to 3 decimals` (although values do actually match)
     "test_whitening"
+
+    # Tests fail with dask>=0.11.2
+    # RuntimeError: Not enough arguments provided
+    # Reported in https://github.com/dask/dask-ml/issues/1003
+    "test_basic"
+    "test_hyperband_patience"
+    "test_same_random_state_same_params"
+    "test_search_patience_infeasible_tol"
+    "test_sha_max_iter_and_metadata"
+    "test_warns_decay_rate"
+    "test_warns_decay_rate_wanted"
   ];
 
   __darwinAllowLocalNetworking = true;