MAINT use sklearn estimator tag in our test suite (#577)

Author: glemaitre

imblearn/over_sampling/_random_over_sampler.py

@@ -119,3 +119,9 @@ def _fit_resample(self, X, y):
                     safe_indexing(y, sample_indices), sample_indices)
         return (safe_indexing(X, sample_indices),
                 safe_indexing(y, sample_indices))
+
+    def _more_tags(self):
+        # TODO: remove the str tag once the following PR is merged:
+        # https://github.com/scikit-learn/scikit-learn/pull/14043
+        return {'X_types': ['2darray', 'str', 'string'],
+                'sample_indices': True}

imblearn/under_sampling/_prototype_generation/_cluster_centroids.py

@@ -170,3 +170,6 @@ def _fit_resample(self, X, y):
         y_resampled = np.hstack(y_resampled)
 
         return X_resampled, np.array(y_resampled, dtype=y.dtype)
+
+    def _more_tags(self):
+        return {'sample_indices': False}

imblearn/under_sampling/_prototype_selection/_condensed_nearest_neighbour.py

@@ -220,3 +220,6 @@ def _fit_resample(self, X, y):
             return (safe_indexing(X, idx_under), safe_indexing(y, idx_under),
                     idx_under)
         return safe_indexing(X, idx_under), safe_indexing(y, idx_under)
+
+    def _more_tags(self):
+        return {'sample_indices': True}

imblearn/under_sampling/_prototype_selection/_edited_nearest_neighbours.py

@@ -186,6 +186,9 @@ def _fit_resample(self, X, y):
                     idx_under)
         return safe_indexing(X, idx_under), safe_indexing(y, idx_under)
 
+    def _more_tags(self):
+        return {'sample_indices': True}
+
 
 @Substitution(
     sampling_strategy=BaseCleaningSampler._sampling_strategy_docstring,
@@ -377,6 +380,9 @@ def _fit_resample(self, X, y):
             return X_resampled, y_resampled, self.sample_indices_
         return X_resampled, y_resampled
 
+    def _more_tags(self):
+        return {'sample_indices': True}
+
 
 @Substitution(
     sampling_strategy=BaseCleaningSampler._sampling_strategy_docstring,
@@ -564,3 +570,6 @@ def _fit_resample(self, X, y):
         if self.return_indices:
             return X_resampled, y_resampled, self.sample_indices_
         return X_resampled, y_resampled
+
+    def _more_tags(self):
+        return {'sample_indices': True}

imblearn/under_sampling/_prototype_selection/_instance_hardness_threshold.py

@@ -187,3 +187,6 @@ def _fit_resample(self, X, y):
             return (safe_indexing(X, idx_under), safe_indexing(y, idx_under),
                     idx_under)
         return safe_indexing(X, idx_under), safe_indexing(y, idx_under)
+
+    def _more_tags(self):
+        return {'sample_indices': True}

imblearn/under_sampling/_prototype_selection/_nearmiss.py

@@ -293,3 +293,6 @@ def _fit_resample(self, X, y):
             return (safe_indexing(X, idx_under), safe_indexing(y, idx_under),
                     idx_under)
         return safe_indexing(X, idx_under), safe_indexing(y, idx_under)
+
+    def _more_tags(self):
+        return {'sample_indices': True}

imblearn/under_sampling/_prototype_selection/_neighbourhood_cleaning_rule.py

@@ -204,3 +204,6 @@ def _fit_resample(self, X, y):
                     self.sample_indices_)
         return (safe_indexing(X, self.sample_indices_),
                 safe_indexing(y, self.sample_indices_))
+
+    def _more_tags(self):
+        return {'sample_indices': True}

imblearn/under_sampling/_prototype_selection/_one_sided_selection.py

@@ -189,3 +189,6 @@ def _fit_resample(self, X, y):
         if self.return_indices:
             return (X_cleaned, y_cleaned, self.sample_indices_)
         return X_cleaned, y_cleaned
+
+    def _more_tags(self):
+        return {'sample_indices': True}

imblearn/under_sampling/_prototype_selection/_random_under_sampler.py

@@ -135,3 +135,9 @@ def _fit_resample(self, X, y):
             return (safe_indexing(X, idx_under), safe_indexing(y, idx_under),
                     idx_under)
         return safe_indexing(X, idx_under), safe_indexing(y, idx_under)
+
+    def _more_tags(self):
+        # TODO: remove the str tag once the following PR is merged:
+        # https://github.com/scikit-learn/scikit-learn/pull/14043
+        return {'X_types': ['2darray', 'str', 'string'],
+                'sample_indices': True}

imblearn/under_sampling/_prototype_selection/_tomek_links.py

@@ -166,3 +166,6 @@ def _fit_resample(self, X, y):
                     self.sample_indices_)
         return (safe_indexing(X, self.sample_indices_),
                 safe_indexing(y, self.sample_indices_))
+
+    def _more_tags(self):
+        return {'sample_indices': True}