unnecessary loop for

CyrilJl · Oct 21, 2024 · bb745f8 · bb745f8
1 parent 9784a40
commit bb745f8
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/optimask/_optimask.py b/optimask/_optimask.py
@@ -249,12 +249,7 @@ def compute_to_keep(size, index_with_nan, permutation, split):
         for i in range(split):
             mask[index_with_nan[permutation[i]]] = 1
 
-        count = 0
-        for i in range(size):
-            if mask[i] == 0:
-                count += 1
-
-        result = np.empty(count, dtype=np.uint32)
+        result = np.empty(size - split, dtype=np.uint32)
         idx = 0
         for i in range(size):
             if mask[i] == 0: