Change flat_correct to use results of arithmetic methods

The methods return a new CCDData, leaving the original unchanged.
astropy · Jun 10, 2014 · c123252 · c123252
1 parent 6b135a6
commit c123252
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ccdproc/core.py b/ccdproc/core.py
@@ -464,15 +464,15 @@ def flat_correct(ccd, flat, min_value=None):
     """
     #Use the min_value to replace any values in the flat
     if min_value is not None:
-       flat.data[flat.data < min_value] = min_value
+        flat.data[flat.data < min_value] = min_value
 
     # normalize the flat
-    flat.divide(flat.data.mean())
+    flat_normed = flat.divide(flat.data.mean())
 
     # divide through the flat
-    ccd.divide(flat)
+    flat_corrected = ccd.divide(flat_normed)
 
-    return ccd
+    return flat_corrected
 
 
 def sigma_func(arr):