diff options
| author | scimax <max.kellermeier@hotmail.de> | 2020-08-20 11:49:22 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-08-20 11:49:22 +0200 |
| commit | de7aa73c56d8128e3fa5cca1e99ff352fffdf0c5 (patch) | |
| tree | 24794064bcc0e4d6a93c668a9584a18c57089ccd /numpy/lib/function_base.py | |
| parent | 16e3bec1744c39915e7ec3e490529c01e2cbde7a (diff) | |
| download | numpy-de7aa73c56d8128e3fa5cca1e99ff352fffdf0c5.tar.gz | |
Minor code clean up
Co-authored-by: Eric Wieser <wieser.eric@gmail.com>
Diffstat (limited to 'numpy/lib/function_base.py')
| -rw-r--r-- | numpy/lib/function_base.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/numpy/lib/function_base.py b/numpy/lib/function_base.py index 910c2bc43..3115f7a80 100644 --- a/numpy/lib/function_base.py +++ b/numpy/lib/function_base.py @@ -1482,7 +1482,7 @@ def angle(z, deg=False): return a -def _unwrap_dispatcher(p, discont=None, axis=None, *, interval_size=2*pi): +def _unwrap_dispatcher(p, discont=None, axis=None, *, interval_size=None): return (p,) @@ -1551,11 +1551,13 @@ def unwrap(p, discont=None, axis=-1, *, interval_size=2*pi): slice1[axis] = slice(1, None) slice1 = tuple(slice1) ddmod = mod(dd + interval_size/2, interval_size) - interval_size/2 + # the above line made `ddmod[abs(dd) == interval_size/2] == -interval_size/2`. + # correct these such that `ddmod[abs(dd) == interval_size/2] == sign(dd)*interval_size/2`. _nx.copyto(ddmod, interval_size/2, where=(ddmod == -interval_size/2) & (dd > 0)) ph_correct = ddmod - dd _nx.copyto(ph_correct, 0, where=abs(dd) < discont) up = array(p, copy=True, dtype='d') - up[slice1] = p[slice1] + ph_correct.cumsum(axis) + up[slice1] = p[slice1] + ph_correct.cumsum(axis) return up |