Commit 5ee1aafd authored by Marcus Wirtz's avatar Marcus Wirtz
Browse files

[coord] Squeuze n-axis in equatorial_scrambling() if n=1

parent e019b095
Pipeline #250175 passed with stages
in 6 minutes and 20 seconds
......@@ -614,4 +614,4 @@ def equatorial_scrambling(v, n=1, coord_system='gal'):
if coord_system != 'eq':
v_scrambled = globals()['eq2%s' % coord_system](v_scrambled)
return np.reshape(v_scrambled, (3, n, -1))
return np.reshape(v_scrambled, (3, n, -1)) if n > 1 else np.reshape(v_scrambled, (3, -1))
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment