unwrap: Move imports to module level.

This commit is contained in:
Jostein Bø Fløystad
2013-11-22 10:42:28 +01:00
parent d2fb06fa3b
commit e72e5de06e
+3 -2
View File
@@ -1,5 +1,8 @@
import numpy as np
import _unwrap_2d
import _unwrap_3d
def unwrap(wrapped_array,
wrap_around_axis_0 = False,
@@ -13,13 +16,11 @@ def unwrap(wrapped_array,
wrapped_array_masked = np.ma.asarray(wrapped_array)
unwrapped_array = np.empty_like(wrapped_array_masked.data)
if wrapped_array.ndim == 2:
import _unwrap_2d
_unwrap_2d._unwrap2D(wrapped_array_masked.data,
np.ma.getmaskarray(wrapped_array_masked).astype(np.uint8),
unwrapped_array,
bool(wrap_around_axis_0), bool(wrap_around_axis_1))
elif wrapped_array.ndim == 3:
import _unwrap_3d
_unwrap_3d._unwrap3D(wrapped_array_masked.data,
np.ma.getmaskarray(wrapped_array_masked).astype(np.uint8),
unwrapped_array,