Added feedback 1

GalagaBits · Mar 17, 2024 · 91e6758 · 91e6758
1 parent b716e87
commit 91e6758
Showing 1 changed file with 1 addition and 10 deletions.
diff --git a/src/spelunker.py b/src/spelunker.py
@@ -284,16 +284,7 @@ def negative_flux_preprocessing(self, fg_array, fg_time, fg_flux):
 
         '''
 
-        neg_elements = []
-        for idx ,i in enumerate(fg_flux):
-            if i < 0:
-                neg_elements.append(idx)
-
-        idx_array = np.arange(len(fg_flux))
-        bool_mask = np.full(len(fg_flux), True)
-
-        for i in neg_elements:
-            bool_mask[i] = False
+        bool_mask = fg_flux >= 0
 
         fg_array = fg_array[bool_mask]
         fg_time = fg_time[bool_mask]