Bug with pos_group at training

CBICA · Jul 18, 2024 · 8bf60d5 · 8bf60d5
1 parent 66573e8
commit 8bf60d5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/spare_scores/spare_scores.py b/spare_scores/spare_scores.py
@@ -24,7 +24,7 @@ def spare_train(
     df: Union[pd.DataFrame, str],
     to_predict: str,
     model_type: str = "SVM",
-    pos_group: str = "",
+    pos_group: Any = "",
     key_var: str = "",
     data_vars: list = [],
     ignore_vars: list = [],

diff --git a/tests/unit/test_spare_scores.py b/tests/unit/test_spare_scores.py
@@ -76,7 +76,7 @@ def test_spare_train(self):
 
         status, result = result["status"], result["data"]
 
-        metadata = result[1]  # For some reason, this is None
+        metadata = result[1]  
         self.assertTrue(status == "OK")
         self.assertTrue(metadata["mdl_type"] == self.model_fixture[1]["mdl_type"])
         self.assertTrue(metadata["kernel"] == self.model_fixture[1]["kernel"])