FIX: fixing error on temporal decoding

robbisg · Jul 26, 2024 · 1093936 · 1093936
1 parent aa6c740
commit 1093936
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/sekupy/analysis/tests/test_decoding.py b/sekupy/analysis/tests/test_decoding.py
@@ -34,7 +34,9 @@ def test_temporal_decoding(fetch_ds):
 
     roi_result = scores['mask-brain_value-2.0']
     assert len(roi_result) == n_permutation + 1
-    assert roi_result[0]['test_score'].shape == (n_splits, 3, 3)
+
+    test_results = np.array(roi_result[0]['test_score'])
+    assert test_results.shape == (n_splits, 3, 3)
     assert np.max(roi_result[0]['test_score']) <= 1.
 
 

diff --git a/sekupy/ext/sklearn/_validation.py b/sekupy/ext/sklearn/_validation.py
@@ -357,7 +357,7 @@ def cross_validate(
     cv = check_cv(cv, y, classifier=is_classifier(estimator))
 
     scorers = check_scoring(
-        estimator, scoring=scoring, #raise_exc=(error_score == "raise")
+        estimator, scoring=scoring, allow_none=(error_score == "raise")
     )
 
     if _routing_enabled():
@@ -1048,8 +1048,7 @@ def _score(estimator, X_test, y_test, scorer, score_params, error_score="raise")
             with suppress(ValueError):
                 # e.g. unwrap memmapped scalars
                 scores = scores.item()
-        if not isinstance(scores, numbers.Number):
-            raise ValueError(error_msg % (scores, type(scores), scorer))
+
     return scores