fix variables

damianhorna · May 8, 2023 · ab84c00 · ab84c00
1 parent 53da89e
commit ab84c00
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/multi_imbalance/resampling/imblearn_wrapper.py b/multi_imbalance/resampling/imblearn_wrapper.py
@@ -40,7 +40,7 @@ def _fit_resample(self, X: np.ndarray, y: np.ndarray) -> Tuple[np.ndarray, np.nd
                 safe_X, safe_y = X_copy[~index], y_copy[~index]
                 current_X, current_y = X_copy[index], y_copy[index]
                 res_X, res_y = self.sampler.fit_resample(current_X, current_y)
-                myX, myy = np.concatenate([res_X, safe_X]), np.concatenate([res_y, safe_y])
+                X_copy, y_copy = np.concatenate([res_X, safe_X]), np.concatenate([res_y, safe_y])
 
             return X_copy, y_copy