Fix indexing with int array

astropy · Jun 13, 2016 · aa8571d · aa8571d
1 parent e8b1397
commit aa8571d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/astroscrappy/astroscrappy.pyx b/astroscrappy/astroscrappy.pyx
@@ -338,7 +338,8 @@ def detect_cosmics(indat, inmask=None, float sigclip=4.5, float sigfrac=0.3,
         # otherwise clean the image and iterate
         if cleantype == 'median':
             # Unmasked median filter
-            cleanarr[crmask] = m5[crmask]
+            cindices = crmask.nonzero()
+            cleanarr[cindices] = m5[cindices]
             del m5
         # Masked mean filter
         elif cleantype == 'meanmask':