Brought changes in line with flake8 formatting

Author: mata

Makefile

@@ -1,7 +1,7 @@
 .PHONY: tests doc build
 
 lint:	
-	flake8 . --exclude=doc
+	flake8 . --exclude=doc mapieenv
 
 type-check:
 	mypy mapie

mapie/classification.py

@@ -178,7 +178,13 @@ class MapieClassifier(BaseEstimator, ClassifierMixin):
      [False False  True]]
     """
 
-    valid_methods_ = ["naive", "score", "cumulated_score", "top_k", "raps","mondrian"]
+    valid_methods_ = ["naive",
+                      "score",
+                      "cumulated_score",
+                      "top_k",
+                      "raps",
+                      "mondrian"
+                      ]
     fit_attributes = [
         "single_estimator_",
         "estimators_",
@@ -1241,7 +1247,7 @@ def predict(
         check_alpha_and_n_samples(alpha_np, n)
 
         if self.method == "naive":
-            self.quantiles_ = 1 - alpha_np 
+            self.quantiles_ = 1 - alpha_np
         else:
             if (cv == "prefit") or (agg_scores in ["mean"]):
                 if self.method == "raps":
@@ -1278,15 +1284,15 @@ def predict(
                     self.quantiles_ = compute_quantiles(
                         self.conformity_scores_,
                         alpha_np,
-                        mondrian = True
+                        mondrian=True
                     )
                 else:
-                    
+
                     self.quantiles_ = compute_quantiles(
                         self.conformity_scores_,
                         alpha_np
                     )
-                    
+
             else:
                 self.quantiles_ = (n + 1) * (1 - alpha_np)
 
@@ -1379,15 +1385,14 @@ def predict(
                 - y_pred_proba_last,
                 -EPSILON
             )
-        elif self.method == "mondrian": #TODO:
-            self.quantiles_ = np.transpose(self.quantiles_,[1,0])
+        elif self.method == "mondrian":
+            self.quantiles_ = np.transpose(self.quantiles_, [1, 0])
 
             prediction_sets = np.greater_equal(
-            y_pred_proba - (1 - self.quantiles_), -EPSILON
+                y_pred_proba - (1 - self.quantiles_),
+                -EPSILON
             )
-            
-            
-            
+
         else:
             raise ValueError(
                 "Invalid method. "

mapie/tests/test_classification.py

@@ -23,7 +23,7 @@
 from mapie._typing import ArrayLike, NDArray
 
 
-METHODS = ["score", "cumulated_score", "raps"]
+METHODS = ["score", "cumulated_score", "raps","mondrian"]
 WRONG_METHODS = ["scores", "cumulated", "test", "", 1, 2.5, (1, 2)]
 WRONG_INCLUDE_LABELS = ["randomised", "True", "False", "other", 1, 2.5, (1, 2)]
 Y_PRED_PROBA_WRONG = [
@@ -264,6 +264,17 @@
             agg_scores="mean"
         )
     ),
+    "mondrian": (
+        Params(
+            method="mondrian",
+            cv="prefit",
+            random_state=None
+        ),
+        ParamsPredict(
+            include_last_label="randomized",
+            agg_scores="mean"
+        )
+    ),
 }
 
 COVERAGES = {