Fix tests

mind-inria · Oct 25, 2024 · 4065a7a · 4065a7a
1 parent 78237c2
commit 4065a7a
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/hidimstat/test/test_cpi.py b/hidimstat/test/test_cpi.py
@@ -23,7 +23,7 @@ def test_cpi(linear_scenario):
         estimator=regression_model,
         imputation_model=imputation_model,
         n_permutations=20,
-        score_proba=False,
+        method="predict",
         random_state=0,
         n_jobs=1,
     )
@@ -55,7 +55,7 @@ def test_cpi(linear_scenario):
         estimator=regression_model,
         imputation_model=imputation_model_list,
         n_permutations=20,
-        score_proba=False,
+        method="predict",
         random_state=0,
         n_jobs=1,
     )
@@ -82,7 +82,7 @@ def test_cpi(linear_scenario):
         n_permutations=20,
         random_state=0,
         n_jobs=1,
-        score_proba=True,
+        method="predict_proba",
         loss=log_loss,
     )
     cpi.fit(

diff --git a/hidimstat/test/test_loco.py b/hidimstat/test/test_loco.py
@@ -20,7 +20,7 @@ def test_loco(linear_scenario):
 
     loco = LOCO(
         estimator=regression_model,
-        score_proba=False,
+        method="predict",
         random_state=0,
         n_jobs=1,
     )
@@ -49,7 +49,7 @@ def test_loco(linear_scenario):
     regression_model.fit(X_train_df, y_train)
     loco = LOCO(
         estimator=regression_model,
-        score_proba=False,
+        method="predict",
         random_state=0,
         n_jobs=1,
     )
@@ -71,7 +71,7 @@ def test_loco(linear_scenario):
 
     loco_clf = LOCO(
         estimator=logistic_model,
-        score_proba=True,
+        method="predict_proba",
         random_state=0,
         n_jobs=1,
         loss=log_loss,

diff --git a/hidimstat/test/test_permutation_importance.py b/hidimstat/test/test_permutation_importance.py
@@ -20,7 +20,7 @@ def test_permutation_importance(linear_scenario):
     pi = PermutationImportance(
         estimator=regression_model,
         n_permutations=20,
-        score_proba=False,
+        method="predict",
         random_state=0,
         n_jobs=1,
     )
@@ -50,7 +50,7 @@ def test_permutation_importance(linear_scenario):
     pi = PermutationImportance(
         estimator=regression_model,
         n_permutations=20,
-        score_proba=False,
+        method="predict",
         random_state=0,
         n_jobs=1,
     )
@@ -73,7 +73,7 @@ def test_permutation_importance(linear_scenario):
     pi_clf = PermutationImportance(
         estimator=logistic_model,
         n_permutations=20,
-        score_proba=True,
+        method="predict_proba",
         random_state=0,
         n_jobs=1,
         loss=log_loss,