From 095964f7ae51e27c97032f69ce38ab9f8d949dab Mon Sep 17 00:00:00 2001
From: "Edgar P. Burkhart" <git@edgarpierre.fr>
Date: Thu, 17 Mar 2022 11:57:29 +0100
Subject: [PATCH] Removed reshape in layers

---
 swash/processing/layers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/swash/processing/layers.py b/swash/processing/layers.py
index dfe41d5..d1656b4 100644
--- a/swash/processing/layers.py
+++ b/swash/processing/layers.py
@@ -50,7 +50,6 @@ fig, ax = plt.subplots()
 # )
 
 n = 0
-velk = velk.reshape((6001, 10, 2, 1251))
 vk = np.sqrt((velk[n] ** 2).sum(axis=1))
 # print(vk.shape)
 # plt.imshow(vk)
@@ -83,6 +82,7 @@ def animate(k):
         )
     for i, l in enumerate(lines):
         l.set_ydata(zk[k, i])
+    return *quiv, *lines
 
 
 ani = animation.FuncAnimation(