Merge pull request soft-matter#209 from caspervdw/pyfftw-fix

FIX: Change datatype instead of copying array
fayazr · Mar 9, 2015 · b97522d · b97522d
2 parents df79f3c + a7302f1
commit b97522d
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/trackpy/preprocessing.py b/trackpy/preprocessing.py
@@ -30,9 +30,7 @@ def fftn(a):
                          "*much* faster.")
             planned = True
         a = pyfftw.n_byte_align(a, a.dtype.alignment)
-        result = np.empty_like(a, dtype=np.complex128)
-        result[:] = pyfftw.interfaces.numpy_fft.fftn(a)
-        return result
+        return pyfftw.interfaces.numpy_fft.fftn(a).astype(np.complex128)
 
     def ifftn(a):
         a = pyfftw.n_byte_align(a, a.dtype.alignment)