From ea8f7c0548f6ed79e07dc1530c0559ed0e1ffd73 Mon Sep 17 00:00:00 2001
From: Carl Buchholz <32228189+aGuyLearning@users.noreply.github.com>
Date: Wed, 5 Feb 2025 17:28:23 +0100
Subject: [PATCH] added expected error

---
 tests/preprocessing/test_imputation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/preprocessing/test_imputation.py b/tests/preprocessing/test_imputation.py
index d9382084..476b1ae0 100644
--- a/tests/preprocessing/test_imputation.py
+++ b/tests/preprocessing/test_imputation.py
@@ -307,7 +307,7 @@ def test_miceforest_impute_numerical_data(impute_iris_adata):
     "array_type,expected_error",
     [
         (np.array, None),
-        (da.array, None),
+        (da.array, NotImplementedError),
         (sparse.csr_matrix, NotImplementedError),
     ],
 )