symprod special case fix

analyticalmindsltd · Nov 25, 2023 · ca2d4e1 · ca2d4e1
1 parent cdf712d
commit ca2d4e1
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/smote_variants/oversampling/_symprod.py b/smote_variants/oversampling/_symprod.py
@@ -162,12 +162,9 @@ def normalize(self, array):
         Returns:
             np.array: the normalized array
         """
-        if (array.max() - array.min()) == 0.0:
-            array = np.repeat(0.0, len(array))
-        else:
-            array = (array - array.min()) / (array.max() - array.min())
-
-        return array
+        return (np.repeat(0.0, len(array))
+                    if array.max() - array.min() < 1e-10
+                    else (array - array.min()) / (array.max() - array.min()))
 
     def determine_cf_norms(self, X_maj, X_min, nn_params):
         """